Skip to content

Commit

Permalink
[Pipelinier] Fix mmav3 pipelining (#5844)
Browse files Browse the repository at this point in the history
Make sure we allocate the right number of slices when doing mmav3
pipelining.
  • Loading branch information
ThomasRaoux authored and Jokeren committed Feb 10, 2025
1 parent f2cda34 commit 04b6c94
Showing 1 changed file with 0 additions and 3 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -527,9 +527,6 @@ assignMemoryLayouts(scf::ForOp &forOp,
loadsToPipeline.insert(&op);
LoadInfo loadInfo;
for (auto use : users) {
// By default we will try pipelining with load to registers at the end.
// For mmav3 we can try leaving the operands in shared memory.
bool mmav3Shmem = false;
if (isa<mlir::triton::DotOpInterface>(use)) {
LDBG("set shared encoding with dot user: " << *use);
auto dot = dyn_cast<tt::DotOp>(use);
Expand Down

0 comments on commit 04b6c94

Please sign in to comment.