API接口 |
说明 |
---|---|
(beta)torch_npu.npu.is_initialized |
Torch_npu提供设备相关的部分接口,具体可参考torch.cuda。 |
(beta)torch_npu.npu.init |
|
(beta)torch_npu.npu.get_device_name |
|
(beta)torch_npu.npu.can_device_access_peer |
|
(beta)torch_npu.npu.get_device_properties |
|
(beta)torch_npu.npu.device_of |
|
(beta)torch_npu.npu.current_blas_handle |
|
(beta)torch_npu.npu.set_stream |
|
(beta)torch_npu.npu.set_sync_debug_mode |
|
(beta)torch_npu.npu.get_sync_debug_mode |
|
(beta)torch_npu.npu.utilization |
|
(beta)torch_npu.npu.get_device_capability |
预留接口,暂不支持,接口默认返回None。 |