昇腾社区首页
中文
注册

torch_npu.npu.aclnn.allow_hf32

函数原型

[object Object]

功能说明

设置conv算子是否支持hf32,一个属性值,对aclnn的allow_hf32属性的设置和查询。

参数说明

bool类型,开启和关闭hf32属性的支持。

输出说明

返回bool类型,返回当前allow_hf32是否开启,默认值为True。

支持的型号

  • [object Object] Atlas 训练系列产品[object Object]
  • [object Object] Atlas A2 训练系列产品[object Object]
  • [object Object] Atlas A3 训练系列产品[object Object]
  • [object Object] Atlas 推理系列产品[object Object]

调用示例

[object Object]