Skip to content

Commit a8202a5

Browse files
authored
Fix case where path is not on inverted (#1436)
1 parent 95f545e commit a8202a5

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

src/enzyme_ad/jax/Implementations/StableHLOAutoDiffOpInterfaceImpl.cpp

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2880,6 +2880,9 @@ struct WhileOpEnzymeOpsRemover
28802880
continue;
28812881

28822882
auto pair = inverted.find(N);
2883+
if (pair == inverted.end()) {
2884+
continue;
2885+
}
28832886
for (const auto &NN : pair->second) {
28842887

28852888
revGraph[NN].insert(N);

0 commit comments

Comments
 (0)