int i, r;
 
        amdgpu_amdkfd_device_fini(adev);
+
+       amdgpu_device_set_pg_state(adev, AMD_PG_STATE_UNGATE);
        /* need to disable SMC first */
        for (i = 0; i < adev->num_ip_blocks; i++) {
                if (!adev->ip_blocks[i].status.hw)
                                          adev->ip_blocks[i].version->funcs->name, r);
                                return r;
                        }
-                       amdgpu_gfx_off_ctrl(adev, false);
-                       cancel_delayed_work_sync(&adev->gfx.gfx_off_delay_work);
+
                        r = adev->ip_blocks[i].version->funcs->hw_fini((void *)adev);
                        /* XXX handle errors */
                        if (r) {
        if (amdgpu_sriov_vf(adev))
                amdgpu_virt_request_full_gpu(adev, false);
 
+       amdgpu_device_set_pg_state(adev, AMD_PG_STATE_UNGATE);
+
        for (i = adev->num_ip_blocks - 1; i >= 0; i--) {
                if (!adev->ip_blocks[i].status.valid)
                        continue;
                DRM_ERROR("set_clockgating_state(ungate) SMC failed %d\n", r);
        }
 
-       /* call smu to disable gfx off feature first when suspend */
-       amdgpu_gfx_off_ctrl(adev, false);
-       cancel_delayed_work_sync(&adev->gfx.gfx_off_delay_work);
-
        for (i = adev->num_ip_blocks - 1; i >= 0; i--) {
                if (!adev->ip_blocks[i].status.valid)
                        continue;
 
        struct amdgpu_device *adev = (struct amdgpu_device *)handle;
        int i;
 
-       amdgpu_device_ip_set_powergating_state(adev, AMD_IP_BLOCK_TYPE_GFX,
-                                              AMD_PG_STATE_UNGATE);
-
        amdgpu_irq_put(adev, &adev->gfx.priv_reg_irq, 0);
        amdgpu_irq_put(adev, &adev->gfx.priv_inst_irq, 0);
 
 
        switch (adev->asic_type) {
        case CHIP_RAVEN:
+               if (!enable) {
+                       amdgpu_gfx_off_ctrl(adev, false);
+                       cancel_delayed_work_sync(&adev->gfx.gfx_off_delay_work);
+               }
                if (adev->pg_flags & AMD_PG_SUPPORT_RLC_SMU_HS) {
                        gfx_v9_0_enable_sck_slow_down_on_power_up(adev, true);
                        gfx_v9_0_enable_sck_slow_down_on_power_down(adev, true);
                /* update mgcg state */
                gfx_v9_0_update_gfx_mg_power_gating(adev, enable);
 
-               /* set gfx off through smu */
-               amdgpu_gfx_off_ctrl(adev, true);
+               if (enable)
+                       amdgpu_gfx_off_ctrl(adev, true);
                break;
        case CHIP_VEGA12:
-               /* set gfx off through smu */
-               amdgpu_gfx_off_ctrl(adev, true);
+               if (!enable) {
+                       amdgpu_gfx_off_ctrl(adev, false);
+                       cancel_delayed_work_sync(&adev->gfx.gfx_off_delay_work);
+               } else {
+                       amdgpu_gfx_off_ctrl(adev, true);
+               }
                break;
        default:
                break;