diff --git a/drivers/gpu/drm/amd/dkms/m4/kthread_use_mm.m4 b/drivers/gpu/drm/amd/dkms/m4/kthread_use_mm.m4
index 0b62fc9008c6b..6177b3b6fa49e 100644
--- a/drivers/gpu/drm/amd/dkms/m4/kthread_use_mm.m4
+++ b/drivers/gpu/drm/amd/dkms/m4/kthread_use_mm.m4
@@ -3,9 +3,21 @@ 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])
+	AC_KERNEL_DO_BACKGROUND([
+		dnl #
+		dnl # sle sp2 server distro inlines kthread_use_mm/kthread_unuse_mm
+		dnl # in mmu_context.h
+		dnl #
+		AC_KERNEL_TRY_COMPILE([
+			#include <linux/mm_types.h>
+			#include <linux/kthread.h>
+			#include <linux/mmu_context.h>
+		], [
+			kthread_use_mm(NULL);
+			kthread_unuse_mm(NULL);
+		], [
+			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 73539fa32b8b1..0a7967f7b8a60 100644
--- a/include/kcl/kcl_kthread.h
+++ b/include/kcl/kcl_kthread.h
@@ -4,9 +4,7 @@
 
 #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);