File tree Expand file tree Collapse file tree 2 files changed +2
-2
lines changed
include/TritonAMDGPUTransforms
lib/TritonAMDGPUTransforms Expand file tree Collapse file tree 2 files changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -38,7 +38,7 @@ unsigned getCyclesPerMfma(DotOp dotOp) {
3838 auto nDim = mfmaLayout.getNDim ();
3939 const auto kDimOperandSize = aTensorTy.getShape ().back ();
4040 // auto kDim = mfmaLayout.getKDim();
41- auto mfmaVersion = mfmaLayout.getVersionMajor ();
41+ auto mfmaVersion = mfmaLayout.getVersion ();
4242 bool allowXF32 =
4343 dotOp.getInputPrecision () == InputPrecision::TF32 && mfmaVersion == 3 ;
4444
Original file line number Diff line number Diff line change @@ -214,7 +214,7 @@ struct DotOpMFMAConverter {
214214 shapeC[1 ] / warpsPerCTA[1 ],
215215 shapeA[1 ],
216216 };
217- auto mfmaVersion = mfmaLayout.getVersionMajor ();
217+ auto mfmaVersion = mfmaLayout.getVersion ();
218218 bool allowXF32 =
219219 dotOp.getInputPrecision () == InputPrecision::TF32 && mfmaVersion == 3 ;
220220
You can’t perform that action at this time.
0 commit comments