Merge 1c9f5960f3 into 6665515349

2026-07-21 23:41:28 +08:00 · 2026-07-20 13:41:51 +08:00
parent 6665515349 1c9f5960f3
commit e5c843a200
1 changed files with 1 additions and 0 deletions
@@ -137,6 +137,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