torch_npu.npu
(beta)torch_npu.npu.get_npu_overflow_flag
(beta)torch_npu.npu.clear_npu_overflow_flag
torch_npu.npu.matmul.allow_hf32
torch_npu.npu.conv.allow_hf32
(beta)torch_npu.npu.set_option
(beta)torch_npu.npu.config.allow_internal_format
(beta)torch_npu.npu.stress_detect
(beta)torch_npu.npu.stop_device
(beta)torch_npu.npu.restart_device
(beta)torch_npu.npu.check_uce_in_memory
torch_npu.npu.SyncLaunchStream
(beta)torch_npu.npu.utils.is_support_inf_nan
(beta)torch_npu.npu.utils.npu_check_overflow
(beta)torch_npu.npu.Event().recorded_time ()
(beta)torch_npu.npu.set_dump
(beta)torch_npu.npu.init_dump
(beta)torch_npu.npu.finalize_dump
(beta)torch_npu.npu.set_compile_mode
(beta)torch_npu.npu.is_jit_compile_false
(beta)torch_npu.npu.set_mm_bmm_format_nd
(beta)torch_npu.npu.get_mm_bmm_format_nd
父主题:
torch_npu.npu