[mlir][sparse] fix mismatch between enter/exitWhileLoop

Peiming Liu · Peiming Liu · commit dcf646a4860c · 2024-01-25T20:04:09.000Z
diff --git a/mlir/lib/Dialect/SparseTensor/Transforms/Utils/LoopEmitter.cpp b/mlir/lib/Dialect/SparseTensor/Transforms/Utils/LoopEmitter.cpp
@@ -599,7 +599,7 @@ Operation *LoopEmitter::enterCoIterationOverTensorsAtLvls(
 
   // NOTE: we can also prepare for next dim here in advance
   // Pushes the loop into stack.
-  loopStack.emplace_back(tidLvls, l, builder.getInsertionBlock(), iv, loopTag);
+  loopStack.emplace_back(tls, l, builder.getInsertionBlock(), iv, loopTag);
   return l;
 }
 

Original file line number	Diff line number	Diff line change
`@@ -599,7 +599,7 @@ Operation *LoopEmitter::enterCoIterationOverTensorsAtLvls(`
`599`	`599`
`600`	`600`	`// NOTE: we can also prepare for next dim here in advance`
`601`	`601`	`// Pushes the loop into stack.`
`602`		`- loopStack.emplace_back(tidLvls, l, builder.getInsertionBlock(), iv, loopTag);`
	`602`	`+ loopStack.emplace_back(tls, l, builder.getInsertionBlock(), iv, loopTag);`
`603`	`603`	`return l;`
`604`	`604`	`}`
`605`	`605`