msm: kgsl: Set snapshot_atomic only after passing all checks

Ensure that the device is powered on and memory is successfully
allocated before setting device->snapshot_atomic to true.

Change-Id: Id11b13e51acb26e024cbed7d8e58940c6e0292a2
Signed-off-by: Rohan Sethi <rohsethi@codeaurora.org>