Closed nrsatish closed 2 years ago
This is to ensure GPU runs can be done in TF32 precision. Adds flags to linear and gemm ubenches as well as DLRM OOTB bench.
This is to ensure GPU runs can be done in TF32 precision. Adds flags to linear and gemm ubenches as well as DLRM OOTB bench.