昇腾社区首页
中文
注册

torch_npu.npu.matmul.allow_hf32

torch_npu.npu.matmul.allow_hf32功能和调用方式与torch.backends.cuda.matmul.allow_tf32类似,torch.backends.cuda.matmul.allow_tf32的功能具体请参考undefined

torch_npu.npu.matmul.allow_hf32的调用方式如下所示:

函数原型

[object Object]

功能说明

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

参数说明

输入bool值,默认值False。

输出说明

bool类型。

支持的型号

  • [object Object] Atlas 训练系列产品[object Object]
  • [object Object] Atlas A2 训练系列产品[object Object]
  • [object Object] Atlas A3 训练系列产品[object Object]
  • [object Object] Atlas 推理系列产品[object Object]

调用示例

[object Object]