(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。
|