kgsl: Modified to work with current kernel's PM