torch.cuda

1

torch.cuda.can_device_access_peer

torch_npu.npu.can_device_access_peer

否

2

torch.cuda.current_blas_handle

torch_npu.npu.current_blas_handle

否

3

torch.cuda.current_device

torch_npu.npu.current_device

是

4

torch.cuda.current_stream

torch_npu.npu.current_stream

是

5

torch.cuda.default_stream

torch_npu.npu.default_stream

是

6

torch.cuda.device

torch_npu.npu.device

是

7

torch.cuda.device_count

torch_npu.npu.device_count

是

8

torch.cuda.device_of

torch_npu.npu.device_of

是

9

torch.cuda.get_arch_list

torch_npu.npu.get_arch_list

否

10

torch.cuda.get_device_capability

torch_npu.npu.get_device_capability

否

11

torch.cuda.get_device_name

torch_npu.npu.get_device_name

否

12

torch.cuda.get_device_properties

torch_npu.npu.get_device_properties

否

13

torch.cuda.get_gencode_flags

torch_npu.npu.get_gencode_flags

否

14

torch.cuda.get_sync_debug_mode

torch_npu.npu.get_sync_debug_mode

否

15

torch.cuda.init

torch_npu.npu.init

是

16

torch.cuda.ipc_collect

torch_npu.npu.ipc_collect

否

17

torch.cuda.is_available

torch_npu.npu.is_available

是

18

torch.cuda.is_initialized

torch_npu.npu.is_initialized

是

19

torch.cuda.memory_usage

torch_npu.npu.memory_usage

否

20

torch.cuda.set_device

torch_npu.npu.set_device

是

21

torch.cuda.set_stream

torch_npu.npu.set_stream

否

22

torch.cuda.set_sync_debug_mode

torch_npu.npu.set_sync_debug_mode

否

23

torch.cuda.stream

torch_npu.npu.stream

是

24

torch.cuda.StreamContext

torch_npu.npu.StreamContext

否

25

torch.cuda.synchronize

torch_npu.npu.synchronize

是

26

torch.cuda.utilization

torch_npu.npu.utilization

否

1

torch.cuda.get_rng_state

torch_npu.npu.get_rng_state

是

2

torch.cuda.get_rng_state_all

torch_npu.npu.get_rng_state_all

是

3

torch.cuda.initial_seed

torch_npu.npu.initial_seed

是

4

torch.cuda.manual_seed

torch_npu.npu.manual_seed

是

5

torch.cuda.manual_seed_all

torch_npu.npu.manual_seed_all

是

6

torch.cuda.seed

torch_npu.npu.seed

是

7

torch.cuda.seed_all

torch_npu.npu.seed_all

是

8

torch.cuda.set_rng_state

torch_npu.npu.set_rng_state

是

9

torch.cuda.set_rng_state_all

torch_npu.npu.set_rng_state_all

是

1

torch.cuda.comm.broadcast

torch_npu.npu.comm.broadcast

否

2

torch.cuda.comm.broadcast_coalesced

torch_npu.npu.comm.broadcast_coalesced

否

3

torch.cuda.comm.gather

torch_npu.npu.comm.gather

否

4

torch.cuda.comm.reduce_add

torch_npu.npu.comm.reduce_add

否

5

torch.cuda.comm.scatter

torch_npu.npu.comm.scatter

否

1

torch.cuda.Event

torch_npu.npu.Event

是，不支持from_ipc_handle和ipc_handle

2

torch.cuda.ExternalStream

torch_npu.npu.ExternalStream

否

3

torch.cuda.Stream

torch_npu.npu.Stream

是

1

torch.cuda.CUDAGraph

torch_npu.npu.CUDAGraph

否

2

torch.cuda.graph

torch_npu.npu.graph

否

3

torch.cuda.graph_pool_handle

torch_npu.npu.graph_pool_handle

否

4

torch.cuda.make_graphed_callables

torch_npu.npu.make_graphed_callables

否

1

torch.cuda.caching_allocator_alloc

torch_npu.npu.caching_allocator_alloc

是

2

torch.cuda.caching_allocator_delete

torch_npu.npu.caching_allocator_delete

是

3

torch.cuda.empty_cache

torch_npu.npu.empty_cache

是

4

torch.cuda.list_gpu_processes

torch_npu.npu.list_gpu_processes

否

5

torch.cuda.max_memory_allocated

torch_npu.npu.max_memory_allocated

是

6

torch.cuda.max_memory_cached

torch_npu.npu.max_memory_cached

是

7

torch.cuda.max_memory_reserved

torch_npu.npu.max_memory_reserved

是

8

torch.cuda.mem_get_info

torch_npu.npu.mem_get_info

否

9

torch.cuda.memory_allocated

torch_npu.npu.memory_allocated

是

10

torch.cuda.memory_cached

torch_npu.npu.memory_cached

是

11

torch.cuda.memory_reserved

torch_npu.npu.memory_reserved

是

12

torch.cuda.memory_snapshot

torch_npu.npu.memory_snapshot

是

13

torch.cuda.memory_stats

torch_npu.npu.memory_stats

是

14

torch.cuda.memory_summary

torch_npu.npu.memory_summary

是

15

torch.cuda.reset_max_memory_allocated

torch_npu.npu.reset_max_memory_allocated

是

16

torch.cuda.reset_max_memory_cached

torch_npu.npu.reset_max_memory_cached

是

17

torch.cuda.reset_peak_memory_stats

torch_npu.npu.reset_peak_memory_stats

是

18

torch.cuda.set_per_process_memory_fraction

torch_npu.npu.set_per_process_memory_fraction

否

1

torch.cuda.nvtx.mark

torch_npu.npu.nvtx.mark

否

2

torch.cuda.nvtx.range_push

torch_npu.npu.nvtx.range_push

否

3

torch.cuda.nvtx.range_pop

torch_npu.npu.nvtx.range_pop

否