torch_npu.npu.aclnn.allow_hf32
接口原型
torch_npu.npu.aclnn.allow_hf32:bool
功能描述
设置conv算子是否支持hf32,一个属性值,对aclnn的allow_hf32属性的设置和查询,默认值为true。
参数说明
aclnn的bool类的属性。
支持的型号
Atlas A2 训练系列产品
调用示例
>>>res = torch.npu.aclnn.allow_hf32 res=True >>>torch.npu.aclnn.allow_hf32 = True >>>res = torch.npu.aclnn.allow_hf32 res=True >>>torch.npu.aclnn.allow_hf32 = False >>>res = torch.npu.aclnn.allow_hf32 res=False
父主题: aclnn