Skip to content

Commit 4a22980

Browse files
authored
Fix infinite loop ori disjoint (#1547)
1 parent c432749 commit 4a22980

File tree

2 files changed

+465
-0
lines changed

2 files changed

+465
-0
lines changed

src/enzyme_ad/jax/Passes/AffineCFG.cpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -518,6 +518,7 @@ AffineApplyNormalizer::AffineApplyNormalizer(AffineMap map,
518518
t.getDefiningOp<ConstantIntOp>() ||
519519
t.getDefiningOp<ConstantIndexOp>())) ||
520520
((decast.getDefiningOp<AddIOp>() || decast.getDefiningOp<SubIOp>() ||
521+
(decast.getDefiningOp<OrIOp>() && isDisjoint(decast)) ||
521522
decast.getDefiningOp<MulIOp>() || decast.getDefiningOp<RemUIOp>() ||
522523
decast.getDefiningOp<RemSIOp>() || decast.getDefiningOp<ShRUIOp>() ||
523524
decast.getDefiningOp<ShLIOp>()) &&

0 commit comments

Comments
 (0)