From d95960b7ce22993cdc119c3e5c4f092a706d3e1c Mon Sep 17 00:00:00 2001
From: Lei Zhenyuan <zhenyuan.lei@intel.com>
Date: Mon, 23 Jun 2025 19:47:34 +0800
Subject: [PATCH] [5/N] Enable intel GPU for unsloth (#2768)

* add is_big_gpu support for xpu

* make code unsloth's style
---
 unsloth/models/_utils.py | 23 +++++++++++++++++++----
 1 file changed, 19 insertions(+), 4 deletions(-)

diff --git a/unsloth/models/_utils.py b/unsloth/models/_utils.py
index c6156fa468..e40635222e 100644
--- a/unsloth/models/_utils.py
+++ b/unsloth/models/_utils.py
@@ -534,13 +534,28 @@ UNSLOTH_COMPILE_MAXIMUM       = os.environ.get("UNSLOTH_COMPILE_MAXIMUM",
 UNSLOTH_COMPILE_IGNORE_ERRORS = os.environ.get("UNSLOTH_COMPILE_IGNORE_ERRORS", "1") == "1"
 # Just remove max_autotune_gemm warning
 import functools
+from torch._inductor.runtime.hints import DeviceProperties
+
+from unsloth import DEVICE_TYPE
+
 @functools.lru_cache(None)
-def is_big_gpu(index):
-    sms = torch.cuda.get_device_properties(index).multi_processor_count
-    if sms < 80:  # V100
-        # log.warning("not enough SMs to use max_autotune_gemm mode")
+def is_big_gpu(index) -> bool:
+
+    if DEVICE_TYPE == "xpu":
+        prop = torch.xpu.get_device_properties(index)
+    else:
+        prop = torch.cuda.get_device_properties(index)
+
+    min_sms = 16 if device.type == "xpu" else 80
+    avail_sms = prop.multi_processor_count
+    if avail_sms < min_sms:
+        log.warning(
+            "Not enough SMs to use max_autotune_gemm mode",
+            extra={"min_sms": min_sms, "avail_sms": avail_sms},
+        )
         return False
     return True
+
 import torch._inductor.utils
 torch._inductor.utils.is_big_gpu = is_big_gpu
 patch_torch_compile(