torch_npu.npu.conv.allow_hf32

torch_npu.npu.conv.allow_hf32功能和调用方式与torch.backends.cudnn.allow_tf32类似，torch.backends.cudnn.allow_tf32的功能具体请参考https://pytorch.org/docs/stable/backends.html#torch.backends.cudnn.allow_tf32。

torch_npu.npu.conv.allow_hf32的调用方式如下所示：

接口原型

torch_npu.npu.conv.allow_hf32 = bool

功能描述

conv类算子开启支持hf32类型能力。

参数说明

输入bool值，默认值True。

输出说明

bool类型。

支持的型号

Atlas 训练系列产品
Atlas A2 训练系列产品
Atlas 推理系列产品

调用示例

>>>import torch
>>>import torch_npu
>>>torch_npu.npu.conv.allow_hf32
True
>>>torch_npu.npu.conv.allow_hf32=False
>>>torch_npu.npu.conv.allow_hf32
False
>>>torch_npu.npu.conv.allow_hf32=True
>>>torch_npu.npu.conv.allow_hf32
True

父主题： torch_npu.npu