概述

（beta）torch_npu.npu.get_npu_overflow_flag

检测npu计算过程中是否有数值溢出。

（beta）torch_npu.npu.clear_npu_overflow_flag

对npu芯片溢出检测为进行清零。

torch_npu.npu.matmul.allow_hf32

功能和调用方式与torch.backends.cuda.matmul.allow_tf32类似。

torch_npu.npu.conv.allow_hf32

功能和调用方式与torch.backends.cudnn.allow_tf32类似。

（beta）torch_npu.npu.set_option

详细使用参见《PyTorch 训练模型迁移调优指南》的“设置算子编译选项”章节。

（beta）torch_npu.npu.config.allow_internal_format

是否使用私有格式，设置为True时允许使用私有格式，设置为False时，不允许申请任何私有格式的tensor，避免了适配层出现私有格式流通。

（beta）torch_npu.npu.get_rng_state

Torch_npu提供随机数相关的部分接口，具体可参考torch.cuda。

（beta）torch_npu.npu.set_rng_state

（beta）torch_npu.npu.get_rng_state_all

（beta）torch_npu.npu.set_rng_state_all

（beta）torch_npu.npu.manual_seed

（beta）torch_npu.npu.manual_seed_all

（beta）torch_npu.npu.seed

（beta）torch_npu.npu.seed_all

（beta）torch_npu.npu.initial_seed

（beta）torch_npu.npu.set_dump

传入配置文件来配置dump参数。

（beta）torch_npu.npu.init_dump

初始化dump配置。

（beta）torch_npu.npu.finalize_dump

结束dump。

（beta）torch_npu.npu.set_compile_mode

设置是否开启二进制。

（beta）torch_npu.npu.is_jit_compile_false

确认算子计算是否采用的二进制，如果是二进制计算，返回True，否则返回False。

（beta）torch_npu.npu.set_mm_bmm_format_nd

设置线性module里面的mm和bmm算子是否用ND格式。

（beta）torch_npu.npu.get_mm_bmm_format_nd

确认线性module里面的mm和bmm算子是否有使能ND格式，如果使能了ND，返回True，否则，返回False。

NPU Device

torch_npu提供设备相关的部分接口。

（beta）torch_npu.npu.caching_allocator_alloc

Torch_npu提供内存管理相关的部分接口，具体可参考torch.cuda。

（beta）torch_npu.npu.caching_allocator_delete

（beta）torch_npu.npu.set_per_process_memory_fraction

（beta）torch_npu.npu.empty_cache

（beta）torch_npu.npu.memory_stats

（beta）torch_npu.npu.memory_stats_as_nested_dict

（beta）torch_npu.npu.reset_accumulated_memory_stats

（beta）torch_npu.npu.reset_peak_memory_stats

（beta）torch_npu.npu.reset_max_memory_allocated

（beta）torch_npu.npu.reset_max_memory_cached

（beta）torch_npu.npu.memory_allocated

（beta）torch_npu.npu.max_memory_allocated

（beta）torch_npu.npu.memory_reserved

（beta）torch_npu.npu.max_memory_reserved

（beta）torch_npu.npu.memory_cached

（beta）torch_npu.npu.max_memory_cached

（beta）torch_npu.npu.memory_snapshot

（beta）torch_npu.npu.memory_summary

（beta）torch_npu.npu.set_aoe

AOE调优使能。

torch_npu.npu.aclnn.version

查询当前aclnn的版本信息,当前版本还没有支持正确的aclnn的版本，兼容返回None同时做warn提示torch.npu.aclnn.version isn't implemented!

torch_npu.npu.aclnn.version

设置conv算子是否支持hf32，一个属性值，对aclnn的allow_hf32属性的设置和查询，默认值为true。

torch.cuda.DoubleTensor

torch_npu.npu.DoubleTensor

是

https://pytorch.org/docs/stable/tensors.html#data-types

torch.cuda.ShortTensor

torch_npu.npu.ShortTensor

是

torch.cuda.CharTensor

torch_npu.npu.CharTensor

是

torch.cuda.ByteTensor

torch_npu.npu.ByteTensor

是

torch.cuda.FloatTensor

torch_npu.npu.FloatTensor

是

torch.cuda.HalfTensor

torch_npu.npu.HalfTensor

是

torch.cuda.IntTensor

torch_npu.npu.IntTensor

是

torch.cuda.BoolTensor

torch_npu.npu.BoolTensor

是

torch.cuda.LongTensor

torch_npu.npu.LongTensor

是