mindspore.device_context.gpu.op_precision.matmul_allow_tf32

查看源文件
mindspore.device_context.gpu.op_precision.matmul_allow_tf32(value)

是否开启矩阵乘在CUBLAS下的TF32张量核计算。 详细信息请查看 nvidia关于CUBLAS_COMPUTE_32F_FAST_TF32的说明

参数:
  • value (bool) - 是否开启矩阵乘在CUBLAS下的TF32张量核计算。如未配置,框架默认为 False