frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Inductor XPU] Add XPU check for is_big_gpu(). (#143491)

Browse Source 
Fix #143472

Pull Request resolved: https://github.com/pytorch/pytorch/pull/143491
Approved by: https://github.com/desertfire, https://github.com/jansel, https://github.com/EikanWang
This commit is contained in:
xinan.lin
2024-12-20 06:23:31 -08:00
committed by PyTorch MergeBot
parent 0da004f3dd
commit af0e159740
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
torch/_inductor/utils.py
View File

@ -1129,7 +1129,7 @@ def is_big_gpu(index_or_device: Union[int, torch.device] = 0) -> bool:
if isinstance(index_or_device, torch.device):
device = index_or_device
else:
device = torch.device("cuda", index_or_device)
device = torch.device(get_gpu_type(), index_or_device)
prop = DeviceProperties.create(device)
@ -1142,7 +1142,7 @@ def is_big_gpu(index_or_device: Union[int, torch.device] = 0) -> bool:
return False
return True
min_sms = 68 # 3080
min_sms = 16 if device.type == "xpu" else 68 # 3080
avail_sms = prop.multi_processor_count
if avail_sms < min_sms:
log.warning(