Skip to content

Commit cdc090a

Browse files
authored
Use SmallPtrSet instead of std::set to make lit test loop-invariant-masks.mlir deterministic (#4323)
Signed-off-by: Anatoly Myachev <[email protected]>
1 parent d5736e4 commit cdc090a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

third_party/intel/lib/Dialect/Triton/Transforms/RemoveMasks.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -444,7 +444,7 @@ class LoopVersioner {
444444
"Expecting a non-empty collection of masked operations");
445445

446446
// Collect the (loop invariant) mask conditions.
447-
std::set<Operation *> maskConds;
447+
SmallPtrSet<Operation *, 8> maskConds;
448448
for (Operation *maskedOp : collector.getMaskedOps()) {
449449
if (auto loadOp = dyn_cast<tt::LoadOp>(maskedOp))
450450
maskConds.insert(loadOp.getMask().getDefiningOp());

0 commit comments

Comments
 (0)