We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 33c3048 commit 70a3f36Copy full SHA for 70a3f36
unsloth/models/_utils.py
@@ -543,16 +543,17 @@ def is_big_gpu(index) -> bool:
543
544
if DEVICE_TYPE == "xpu":
545
prop = torch.xpu.get_device_properties(index)
546
+ min_sms = 16
547
else:
548
prop = torch.cuda.get_device_properties(index)
549
+ min_sms = 80
550
- min_sms = 16 if device.type == "xpu" else 80
551
avail_sms = prop.multi_processor_count
552
if avail_sms < min_sms:
- log.warning(
553
- "Not enough SMs to use max_autotune_gemm mode",
554
- extra={"min_sms": min_sms, "avail_sms": avail_sms},
555
- )
+ #log.warning(
+ # "Not enough SMs to use max_autotune_gemm mode",
+ # extra={"min_sms": min_sms, "avail_sms": avail_sms},
556
+ #)
557
return False
558
return True
559
0 commit comments