* Returns:
  * 0 for success, error for failure.
  */
-int amdgpu_vm_init(struct amdgpu_device *adev, struct amdgpu_vm *vm,
-                  int vm_context, u32 pasid)
+int amdgpu_vm_init(struct amdgpu_device *adev, struct amdgpu_vm *vm, u32 pasid)
 {
        struct amdgpu_bo_param bp;
        struct amdgpu_bo *root;
        vm->pte_support_ats = false;
        vm->is_compute_context = false;
 
-       if (vm_context == AMDGPU_VM_CONTEXT_COMPUTE) {
-               vm->use_cpu_for_update = !!(adev->vm_manager.vm_update_mode &
-                                               AMDGPU_VM_USE_CPU_FOR_COMPUTE);
+       vm->use_cpu_for_update = !!(adev->vm_manager.vm_update_mode &
+                                   AMDGPU_VM_USE_CPU_FOR_GFX);
 
-               if (adev->asic_type == CHIP_RAVEN)
-                       vm->pte_support_ats = true;
-       } else {
-               vm->use_cpu_for_update = !!(adev->vm_manager.vm_update_mode &
-                                               AMDGPU_VM_USE_CPU_FOR_GFX);
-       }
        DRM_DEBUG_DRIVER("VM update mode is %s\n",
                         vm->use_cpu_for_update ? "CPU" : "SDMA");
        WARN_ONCE((vm->use_cpu_for_update &&
        vm->evicting = false;
 
        amdgpu_vm_bo_param(adev, vm, adev->vm_manager.root_level, false, &bp);
-       if (vm_context == AMDGPU_VM_CONTEXT_COMPUTE)
-               bp.flags &= ~AMDGPU_GEM_CREATE_SHADOW;
        r = amdgpu_bo_create(adev, &bp, &root);
        if (r)
                goto error_free_delayed;
 
 void amdgpu_vm_manager_fini(struct amdgpu_device *adev);
 
 long amdgpu_vm_wait_idle(struct amdgpu_vm *vm, long timeout);
-int amdgpu_vm_init(struct amdgpu_device *adev, struct amdgpu_vm *vm,
-                  int vm_context, u32 pasid);
+int amdgpu_vm_init(struct amdgpu_device *adev, struct amdgpu_vm *vm, u32 pasid);
 int amdgpu_vm_make_compute(struct amdgpu_device *adev, struct amdgpu_vm *vm, u32 pasid);
 void amdgpu_vm_release_compute(struct amdgpu_device *adev, struct amdgpu_vm *vm);
 void amdgpu_vm_fini(struct amdgpu_device *adev, struct amdgpu_vm *vm);