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 A3 训练系列产品 Atlas 推理系列产品
调用示例
1 2 3 4 5 6 7 8 9 10 | >>>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