diff --git a/drivers/gpu/drm/amd/dkms/m4/kernel.m4 b/drivers/gpu/drm/amd/dkms/m4/kernel.m4
index 78dbac3de3959..929dcd71fd736 100644
--- a/drivers/gpu/drm/amd/dkms/m4/kernel.m4
+++ b/drivers/gpu/drm/amd/dkms/m4/kernel.m4
@@ -201,6 +201,7 @@ AC_DEFUN([AC_CONFIG_KERNEL], [
 	AC_AMDGPU_DRM_HDCP_UPDATE_CONTENT_PROTECTION
 	AC_AMDGPU_DOWN_READ_KILLABLE
 	AC_AMDGPU_DRM_DRIVER_RELEASE
+	AC_AMDGPU_KTHREAD_USE_MM
 
 	AC_KERNEL_WAIT
 	AS_IF([test "$LINUX_OBJ" != "$LINUX"], [
diff --git a/drivers/gpu/drm/amd/dkms/m4/kthread_use_mm.m4 b/drivers/gpu/drm/amd/dkms/m4/kthread_use_mm.m4
new file mode 100644
index 0000000000000..0b62fc9008c6b
--- /dev/null
+++ b/drivers/gpu/drm/amd/dkms/m4/kthread_use_mm.m4
@@ -0,0 +1,11 @@
+dnl #
+dnl # f5678e7f2ac3 kernel: better document the use_mm/unuse_mm API contract
+dnl # 9bf5b9eb232b kernel: move use_mm/unuse_mm to kthread.c
+dnl #
+AC_DEFUN([AC_AMDGPU_KTHREAD_USE_MM], [
+	AC_KERNEL_CHECK_SYMBOL_EXPORT([kthread_use_mm kthread_unuse_mm],
+	[kernel/kthread.c], [
+		AC_DEFINE(HAVE_KTHREAD_USE_MM, 1,
+			[kthread_{use,unuse}_mm() is available])
+	])
+])
diff --git a/include/kcl/kcl_kthread.h b/include/kcl/kcl_kthread.h
index 8a08425d72312..73539fa32b8b1 100644
--- a/include/kcl/kcl_kthread.h
+++ b/include/kcl/kcl_kthread.h
@@ -4,6 +4,9 @@
 
 #include <linux/sched.h>
 #include <linux/kthread.h>
+#ifndef HAVE_KTHREAD_USE_MM
+#include <linux/mmu_context.h>
+#endif
 
 #if !defined(HAVE___KTHREAD_SHOULD_PATK)
 extern bool __kcl_kthread_should_park(struct task_struct *k);
@@ -15,4 +18,18 @@ extern void (*_kcl_kthread_unpark)(struct task_struct *k);
 extern int (*_kcl_kthread_park)(struct task_struct *k);
 extern bool (*_kcl_kthread_should_park)(void);
 #endif
+
+#ifndef HAVE_KTHREAD_USE_MM
+static inline
+void kthread_use_mm(struct mm_struct *mm)
+{
+	use_mm(mm);
+}
+static inline
+void kthread_unuse_mm(struct mm_struct *mm)
+{
+	unuse_mm(mm);
+}
+#endif
+
 #endif /* AMDKCL_KTHREAD_H */