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

fix ifu gemm perf regression (#348)

Browse Source
This commit is contained in:
Shucai Xiao
2023-10-04 08:45:18 -05:00
committed by GitHub
parent e6f75d05e3
commit 8049891ff7
1 changed files with 7 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
lib/Dialect/TritonGPU/Transforms/RemoveLayoutConversions.cpp
View File

@@ -419,6 +419,13 @@ SmallVector<Value> LayoutPropagation::propagateToUsers(Value value,
user->hasTrait<mlir::OpTrait::Elementwise>() ||
isa<triton::ReduceOp, triton::ExpandDimsOp,
triton::gpu::ConvertLayoutOp>(user)) {
#ifdef USE_ROCM
if (auto convertOp = dyn_cast<triton::gpu::ConvertLayoutOp>(user)) {
if (triton::gpu::isSharedEncoding(convertOp.getResult()) ||
triton::gpu::isSharedEncoding(convertOp.getOperand()))
continue;
}
#endif
setEncoding(user->getResults(), info, changed, user);
continue;
}