Skip to content

Commit 58798c7

Browse files
committed
[AMD] Added patches for upstream changes
1 parent fc22338 commit 58798c7

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

third_party/amd/include/TritonAMDGPUTransforms/DotTiling.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff 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

third_party/amd/lib/TritonAMDGPUTransforms/RefineOps.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff 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

0 commit comments

Comments
 (0)