昇腾社区首页
中文
注册
开发者
下载

torch_npu.npu.aclnn.allow_hf32

产品支持情况

产品 是否支持
[object Object]Atlas A3 训练系列产品[object Object]
[object Object]Atlas A2 训练系列产品[object Object]
[object Object]Atlas 推理系列产品[object Object]
[object Object]Atlas 训练系列产品[object Object]

功能说明

设置或查询conv类算子是否支持hf32。

函数原型

[object Object]

参数说明

bool:开启和关闭hf32属性的支持。

返回值说明

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

调用示例

[object Object]