检查NPU是否可用。
()
| 32 | |
| 33 | |
| 34 | def is_npu_available(): |
| 35 | """检查NPU是否可用。""" |
| 36 | try: |
| 37 | import torch_npu |
| 38 | |
| 39 | return torch_npu.npu.is_available() |
| 40 | except ImportError: |
| 41 | return False |
| 42 | |
| 43 | |
| 44 | def _resolve_ncpu(config, fallback=4): |
no outgoing calls
no test coverage detected
searching dependent graphs…