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
。