API接口 |
说明 |
|---|---|
torch_npu.npu.is_initialized |
Torch_npu提供设备相关的部分接口，具体可参考torch.cuda。 |
torch_npu.npu.init |
|
torch_npu.npu.get_device_name |
|
torch_npu.npu.can_device_access_peer |
|
torch_npu.npu.get_device_properties |
|
torch_npu.npu.get_device_capability |
|
torch_npu.npu.device_of |
|
torch_npu.npu.current_blas_handle |
|
torch_npu.npu.set_stream |
|
torch_npu.npu.set_sync_debug_mode |
|
torch_npu.npu.get_sync_debug_mode |
|
torch_npu.npu.utilization |