/*
         * TODO: bypass the loading in sriov for now
         */
-       if (amdgpu_sriov_vf(psp->adev))
-               return 0;
 
        cmd = kzalloc(sizeof(struct psp_gfx_cmd_resp), GFP_KERNEL);
        if (!cmd)
        /*
         * TODO: bypass the unloading in sriov for now
         */
-       if (amdgpu_sriov_vf(psp->adev))
-               return 0;
 
        cmd = kzalloc(sizeof(struct psp_gfx_cmd_resp), GFP_KERNEL);
        if (!cmd)
        int ret;
        struct psp_gfx_cmd_resp *cmd;
 
-       /*
-        * TODO: bypass the loading in sriov for now
-       */
-       if (amdgpu_sriov_vf(psp->adev))
-               return 0;
 
        cmd = kzalloc(sizeof(struct psp_gfx_cmd_resp), GFP_KERNEL);
        if (!cmd)
        if (!psp->cmd)
                return -ENOMEM;
 
-       /* this fw pri bo is not used under SRIOV */
-       if (!amdgpu_sriov_vf(psp->adev)) {
-               ret = amdgpu_bo_create_kernel(adev, PSP_1_MEG, PSP_1_MEG,
-                                             AMDGPU_GEM_DOMAIN_GTT,
-                                             &psp->fw_pri_bo,
-                                             &psp->fw_pri_mc_addr,
-                                             &psp->fw_pri_buf);
-               if (ret)
-                       goto failed;
-       }
+       ret = amdgpu_bo_create_kernel(adev, PSP_1_MEG, PSP_1_MEG,
+                                       AMDGPU_GEM_DOMAIN_GTT,
+                                       &psp->fw_pri_bo,
+                                       &psp->fw_pri_mc_addr,
+                                       &psp->fw_pri_buf);
+       if (ret)
+               goto failed;
 
        ret = amdgpu_bo_create_kernel(adev, PSP_FENCE_BUFFER_SIZE, PAGE_SIZE,
                                        AMDGPU_GEM_DOMAIN_VRAM,