Add test cases to check translation of the following NVVM ops:-
1.) wmma.m16n16k16.load 2.) wmma.m16n16k1616.store 3.) wmma.m16n16k16.mma
Differential D95333
[MLIR][NVVM] Add test cases to check translation of matrix-multiply accumulate ops to the corresponding intrinsics in NVPTX backend navdeepkk on Jan 24 2021, 11:24 PM. Authored by
Details
Add test cases to check translation of the following NVVM ops:- 1.) wmma.m16n16k16.load 2.) wmma.m16n16k1616.store 3.) wmma.m16n16k16.mma
Diff Detail
Event Timeline
|