torch_npu.npu.matmul.allow_hf32功能和调用方式与torch.backends.cuda.matmul.allow_tf32类似,torch.backends.cuda.matmul.allow_tf32的功能具体请参考https://pytorch.org/docs/stable/backends.html#torch.backends.cuda.matmul.allow_tf32。
torch_npu.npu.matmul.allow_hf32的调用方式如下所示:
torch_npu.npu.matmul.allow_hf32 = bool
matmul类算子开启支持hf32类型能力。
输入bool值,默认值False。
bool类型。
1 2 3 4 5 6 7 8 9 10 | >>>import torch >>>import torch_npu >>>torch_npu.npu.matmul.allow_hf32 False >>>torch_npu.npu.matmul.allow_hf32=True >>>torch_npu.npu.matmul.allow_hf32 True >>>torch_npu.npu.matmul.allow_hf32=False >>>torch_npu.npu.matmul.allow_hf32 False |