Commit fa582c6f authored by Philip Yang's avatar Philip Yang Committed by Alex Deucher
Browse files

drm/amdkfd: Use mmget_not_zero in MMU notifier 



MMU notifier callback may pass in mm with mm->mm_users==0 when process
is exiting, use mmget_no_zero to avoid accessing invalid mm in deferred
list work after mm is gone.

Signed-off-by: default avatarPhilip Yang <Philip.Yang@amd.com>
Reviewed-by: default avatarFelix Kuehling <Felix.Kuehling@amd.com>
Signed-off-by: default avatarAlex Deucher <alexander.deucher@amd.com>
parent 2a460963

drivers/gpu/drm/amd/amdkfd/kfd_svm.c

+3 −0
Original line number Diff line number Diff line
@@ -2307,6 +2307,8 @@ svm_range_cpu_invalidate_pagetables(struct mmu_interval_notifier *mni, 


	if (range->event == MMU_NOTIFY_RELEASE)

		return true;

	if (!mmget_not_zero(mni->mm))

		return true;



	start = mni->interval_tree.start;

	last = mni->interval_tree.last;
@@ -2333,6 +2335,7 @@ svm_range_cpu_invalidate_pagetables(struct mmu_interval_notifier *mni, 
	}



	svm_range_unlock(prange);

	mmput(mni->mm);



	return true;

}