goto out_unlock_prange;
        }
  
 -      r = svm_migrate_vram_to_ram(prange, mm, KFD_MIGRATE_TRIGGER_PAGEFAULT_CPU,
 -                              vmf->page);
 +      r = svm_migrate_vram_to_ram(prange, vmf->vma->vm_mm,
-                                   KFD_MIGRATE_TRIGGER_PAGEFAULT_CPU);
++                                  KFD_MIGRATE_TRIGGER_PAGEFAULT_CPU,
++                                  vmf->page);
        if (r)
 -              pr_debug("failed %d migrate 0x%p [0x%lx 0x%lx] to ram\n", r,
 -                       prange, prange->start, prange->last);
 +              pr_debug("failed %d migrate svms 0x%p range 0x%p [0x%lx 0x%lx]\n",
 +                       r, prange->svms, prange, prange->start, prange->last);
  
        /* xnack on, update mapping on GPUs with ACCESS_IN_PLACE */
        if (p->xnack_enabled && parent == prange)