github/ROCm
1
1
Fork 0
mirror of https://github.com/ROCm/ROCm.git synced 2026-04-27 03:01:52 -04:00
Code Issues Packages Projects Releases Wiki Activity

Disable WMMA dot transformation

Browse Source 
Enabled only in lit test.
Revert after complete enabling WMMA

Signed-off-by: joviliast <iveselov.nn@gmail.com>
This commit is contained in:
joviliast
2023-12-16 00:32:20 +02:00
committed by Lixun Zhang
parent 47e801730c
commit 5c182aa73a
4 changed files with 15 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
lib/Dialect/TritonGPU/Transforms/AccelerateAMDMatmul.cpp
View File

@@ -398,9 +398,11 @@ class TritonAMDGPUAccelerateMatmulPass
public:
TritonAMDGPUAccelerateMatmulPass() = default;
TritonAMDGPUAccelerateMatmulPass(StringRef archGen,
int matrixInstructionSize) {
int matrixInstructionSize,
bool enableWmmaTransform) {
this->archGenerationName = archGen.data();
this->matrixInstructionSize = matrixInstructionSize;
this->enableWmmaTransform = enableWmmaTransform;
}
void runOnOperation() override {
MLIRContext *context = &getContext();
@@ -413,9 +415,9 @@ public:
MatrixCoreVersion::CDNA_MFMA3 == matrixCoreVer) {
patterns.add<::BlockedToMFMA>(context, getMfmaVersion(matrixCoreVer),
matrixInstructionSize);
} /*else if (matrixCoreVer == MatrixCoreVersion::RDNA_WMMA) {
} else if (matrixCoreVer == MatrixCoreVersion::RDNA_WMMA) {
patterns.add<::BlockedToWMMA>(context);
}*/
}
if (applyPatternsAndFoldGreedily(m, std::move(patterns)).failed()) {
signalPassFailure();
}
@@ -424,7 +426,8 @@ public:
std::unique_ptr<Pass>
mlir::createTritonAMDGPUAccelerateMatmulPass(std::string archGen,
int matrixInstructionSize) {
int matrixInstructionSize,
bool enableWmmaTransform) {
return std::make_unique<TritonAMDGPUAccelerateMatmulPass>(
archGen, matrixInstructionSize);
archGen, matrixInstructionSize, enableWmmaTransform);
}