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

[BACKEND] Use our internal slice implementation to avoid combinatoria… (#2535)

Browse Source 
…l explosion
This commit is contained in:
Thomas Raoux
2023-10-23 20:06:34 -07:00
committed by GitHub
parent c65d2c2ed6
commit 5e6071254c
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
lib/Dialect/TritonGPU/Transforms/AccelerateMatmul.cpp
View File

@@ -48,7 +48,7 @@ warpsPerTileV2(tt::DotOp dotOp, const ArrayRef<int64_t> shape, int numWarps) {
auto filter = [&dotOp](Operation *op) {
return op->getParentRegion() == dotOp->getParentRegion();
};
auto slices = mlir::getSlice(dotOp, {filter});
auto slices = multiRootGetSlice(dotOp, {filter});
for (Operation *op : slices)
if (isa<tt::DotOp>(op) && (op != dotOp))
return {(unsigned)numWarps, 1};