- 若API未标明支持情况则代表该API暂无验证结果，待验证后更新表格。
- 在使用支持的cuda接口时，需要将API名称中的cuda变换为NPU形式才能使用：torch.cuda.***变换为torch_npu.npu.***。举例如下：
trch.cuda.current_device --> torch_npu.npu.current_device
API名称 |
NPU形式名称 |
是否支持 |
限制与说明 |
|---|---|---|---|
torch.cuda.can_device_access_peer |
torch_npu.npu.can_device_access_peer |
是 |
|
torch.cuda.current_blas_handle |
torch_npu.npu.current_blas_handle |
是 |
|
torch.cuda.current_device |
torch_npu.npu.current_device |
是 |
|
torch.cuda.device_count |
torch_npu.npu.device_count |
是 |
|
torch.cuda.device_of |
torch_npu.npu.device_of |
是 |
|
torch.cuda.get_device_name |
torch_npu.npu.get_device_name |
是 |
|
torch.cuda.get_device_properties |
torch_npu.npu.get_device_properties |
是 |
返回数据结构不支持major 、minor 、multi_processor_count字段 |
torch.cuda.get_sync_debug_mode |
torch_npu.npu.get_sync_debug_mode |
是 |
|
torch.cuda.init |
torch_npu.npu.init |
是 |
|
torch.cuda.is_available |
torch_npu.npu.is_available |
是 |
|
torch.cuda.is_initialized |
torch_npu.npu.is_initialized |
是 |
|
torch.cuda.set_device |
torch_npu.npu.set_device |
是 |
|
torch.cuda.set_stream |
torch_npu.npu.set_stream |
是 |
|
torch.cuda.set_sync_debug_mode |
torch_npu.npu.set_sync_debug_mode |
是 |
|
torch.cuda.stream |
torch_npu.npu.stream |
是 |
|
torch.cuda.utilization |
torch_npu.npu.utilization |
是 |
|
torch.cuda.get_rng_state |
torch_npu.npu.get_rng_state |
是 |
|
torch.cuda.set_rng_state |
torch_npu.npu.set_rng_state |
是 |
|
torch.cuda.set_rng_state_all |
torch_npu.npu.set_rng_state_all |
是 |
|
torch.cuda.manual_seed |
torch_npu.npu.manual_seed |
是 |
|
torch.cuda.manual_seed_all |
torch_npu.npu.manual_seed_all |
是 |
|
torch.cuda.seed |
torch_npu.npu.seed |
是 |
|
torch.cuda.seed_all |
torch_npu.npu.seed_all |
是 |
|
torch.cuda.initial_seed |
torch_npu.npu.initial_seed |
是 |
|
torch.cuda.Stream |
torch_npu.npu.Stream |
是 |
|
torch.cuda.Stream.wait_stream |
torch_npu.npu.Stream.wait_stream |
是 |
|
torch.cuda.Event |
torch_npu.npu.Event |
是 |
|
torch.cuda.Event.elapsed_time |
torch_npu.npu.Event.elapsed_time |
是 |
|
torch.cuda.Event.query |
torch_npu.npu.Event.query |
是 |
|
torch.cuda.Event.wait |
torch_npu.npu.Event.wait |
是 |
|
torch.cuda.empty_cache |
torch_npu.npu.empty_cache |
是 |
|
torch.cuda.memory_stats |
torch_npu.npu.memory_stats |
是 |
|
torch.cuda.memory_summary |
torch_npu.npu.memory_summary |
是 |
|
torch.cuda.memory_allocated |
torch_npu.npu.memory_allocated |
是 |
|
torch.cuda.max_memory_allocated |
torch_npu.npu.max_memory_allocated |
是 |
|
torch.cuda.reset_max_memory_allocated |
torch_npu.npu.reset_max_memory_allocated |
是 |
|
torch.cuda.memory_reserved |
torch_npu.npu.memory_reserved |
是 |
|
torch.cuda.max_memory_reserved |
torch_npu.npu.max_memory_reserved |
是 |
|
torch.cuda.set_per_process_memory_fraction |
torch_npu.npu.set_per_process_memory_fraction |
是 |
|
torch.cuda.memory_cached |
torch_npu.npu.memory_cached |
是 |
|
torch.cuda.max_memory_cached |
torch_npu.npu.max_memory_cached |
是 |
|
torch.cuda.reset_max_memory_cached |
torch_npu.npu.reset_max_memory_cached |
是 |
|
torch.cuda.reset_peak_memory_stats |
torch_npu.npu.reset_peak_memory_stats |
是 |
|
torch.cuda.caching_allocator_alloc |
torch_npu.npu.caching_allocator_alloc |
是 |
|
torch.cuda.caching_allocator_delete |
torch_npu.npu.caching_allocator_delete |
是 |
|
torch.cuda.StreamContext |
|||
torch.cuda.current_stream |
|||
torch.cuda.default_stream |
|||
torch.cuda.device |
|||
torch.cuda.get_arch_list |
|||
torch.cuda.get_device_capability |
|||
torch.cuda.get_gencode_flags |
|||
torch.cuda.ipc_collect |
|||
torch.cuda.memory_usage |
|||
torch.cuda.synchronize |
|||
torch.cuda.temperature |
|||
torch.cuda.power_draw |
|||
torch.cuda.clock_rate |
|||
torch.cuda.get_rng_state_all |
|||
torch.cuda.comm.broadcast |
|||
torch.cuda.comm.broadcast_coalesced |
|||
torch.cuda.comm.reduce_add |
|||
torch.cuda.comm.scatter |
|||
torch.cuda.comm.gather |
|||
torch.cuda.Stream.query |
|||
torch.cuda.Stream.record_event |
|||
torch.cuda.Stream.synchronize |
|||
torch.cuda.Stream.wait_event |
|||
torch.cuda.ExternalStream |
|||
torch.cuda.ExternalStream.query |
|||
torch.cuda.ExternalStream.record_event |
|||
torch.cuda.ExternalStream.synchronize |
|||
torch.cuda.ExternalStream.wait_event |
|||
torch.cuda.ExternalStream.wait_stream |
|||
torch.cuda.Event.from_ipc_handle |
|||
torch.cuda.Event.ipc_handle |
|||
torch.cuda.Event.record |
|||
torch.cuda.Event.synchronize |
|||
torch.cuda.is_current_stream_capturing |
|||
torch.cuda.graph_pool_handle |
|||
torch.cuda.CUDAGraph |
|||
torch.cuda.CUDAGraph.capture_begin |
|||
torch.cuda.CUDAGraph.capture_end |
|||
torch.cuda.CUDAGraph.debug_dump |
|||
torch.cuda.CUDAGraph.enable_debug_mode |
|||
torch.cuda.CUDAGraph.pool |
|||
torch.cuda.CUDAGraph.replay |
|||
torch.cuda.CUDAGraph.reset |
|||
torch.cuda.graph |
|||
torch.cuda.make_graphed_callables |
|||
torch.cuda.list_gpu_processes |
|||
torch.cuda.mem_get_info |
torch_npu.npu.mem_get_info |
是 |
|
torch.cuda.memory_snapshot |
|||
torch.cuda.get_allocator_backend |
否 |
||
torch.cuda.CUDAPluggableAllocator |
torch.npu.npu.NPUPluggableAllocator |
是 |
该接口涉及高危操作，使用请参考torch.npu.npu.NPUPluggableAllocator。 |
torch.cuda.change_current_allocator |
torch.npu.npu.change_current_allocator |
是 |
该接口涉及高危操作，使用请参考torch.npu.npu.change_current_allocator。 |
torch.cuda.nvtx.mark |
|||
torch.cuda.nvtx.range_push |
|||
torch.cuda.nvtx.range_pop |
|||
torch.cuda.jiterator._create_jit_fn |
|||
torch.cuda.jiterator._create_multi_output_jit_fn |
|||
torch.cuda._sanitizer.enable_cuda_sanitizer |