This is an archive of the discontinued LLVM Phabricator instance.

[mlir][nvvm] Fix support for tf32 data type in mma.sync

Authored by christopherbate on May 4 2022, 7:19 PM.



The NVVM dialect test coverage for all possible type/shape combinations
in the op is mostly complete. However, there were tests
missing for TF32 datatype support. This change adds tests for the one
relevant shape/type combination. This uncovered a small bug in the op
verifier, which this change also fixes.

Diff Detail

Event Timeline

christopherbate created this revision.May 4 2022, 7:19 PM
Herald added a project: Restricted Project. · View Herald Transcript
christopherbate requested review of this revision.May 4 2022, 7:19 PM
nirvedhmeshram accepted this revision.May 4 2022, 7:45 PM
This revision is now accepted and ready to land.May 4 2022, 7:45 PM

Fix type in test name: 'nvvm_mma_m16n8k4_f64_f64' should have been 'nvvm_mma_m16n8k4_tf32_f32'