Ignore icmp usage of sret

wsmoses · wsmoses · commit 5bd506013264 · 2026-01-28T21:33:55.000-06:00
diff --git a/enzyme/Enzyme/CApi.cpp b/enzyme/Enzyme/CApi.cpp
@@ -1766,6 +1766,9 @@ bool needsReRooting(llvm::Argument *arg, bool &anyJLStore,
   for (auto &&[I, cur, byteOffset] : findAllUsersOf(arg)) {
     assert(I->getParent()->getParent() == arg->getParent());
 
+    if (isa<ICmpInst>(I)) {
+      continue;
+    }
     if (isa<LoadInst>(I)) {
       continue;
     }

Original file line number	Diff line number	Diff line change
`@@ -1766,6 +1766,9 @@ bool needsReRooting(llvm::Argument *arg, bool &anyJLStore,`
`1766`	`1766`	`for (auto &&[I, cur, byteOffset] : findAllUsersOf(arg)) {`
`1767`	`1767`	`assert(I->getParent()->getParent() == arg->getParent());`
`1768`	`1768`
	`1769`	`+ if (isa<ICmpInst>(I)) {`
	`1770`	`+ continue;`
	`1771`	`+ }`
`1769`	`1772`	`if (isa<LoadInst>(I)) {`
`1770`	`1773`	`continue;`
`1771`	`1774`	`}`