tps61310: enable wake_lock when flash enabled

Change-Id: I098bd28504039d07b5c64404a0031df6ca1c098c