Merge 5c565e9125b475294422b99d361d24920c55ed91 into 0899012ad60db23cbc5990d164fbd22195bafcb2

This commit is contained in:
Rui Wang (王瑞) 2025-12-20 13:36:36 +01:00 committed by GitHub
commit 69659c2309
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -123,6 +123,7 @@ except:
try:
import torch_npu # noqa: F401
_ = torch.npu.device_count()
torch_npu.npu.set_compile_mode(jit_compile = False)
npu_available = torch.npu.is_available()
except:
npu_available = False