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:
parent
1fc64a0a1b
commit
1b4ca535ae
1 changed files with 2 additions and 5 deletions
|
|
@ -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
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue