NPU Device

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。