torch_npu
（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
NPU Device
父主题：
torch_npu.npu