move min_sms in is_big_gpu inside DEVICE_TYPE if else (#2792)

log is not defined in torch inductor so remove

Remove log.warning entirely
This commit is contained in:
DoubleMathew 2025-06-23 20:57:55 -05:00 committed by GitHub
commit 1b4ca535ae

View file

@ -543,16 +543,13 @@ def is_big_gpu(index) -> bool:
if DEVICE_TYPE == "xpu":
prop = torch.xpu.get_device_properties(index)
min_sms = 16
else:
prop = torch.cuda.get_device_properties(index)
min_sms = 80
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