Fix pgcstack bug (rust-lang#977)

wsmoses · web-flow · commit 5d1e0003ec7b · 2023-02-03T05:42:17.000Z
diff --git a/enzyme/Enzyme/DifferentialUseAnalysis.h b/enzyme/Enzyme/DifferentialUseAnalysis.h
@@ -462,7 +462,7 @@ static inline bool is_value_needed_in_reverse(
   if (auto CI = dyn_cast<CallInst>(inst)) {
     StringRef funcName = getFuncNameFromCall(const_cast<CallInst *>(CI));
     if (funcName == "julia.get_pgcstack" || funcName == "julia.ptls_states")
-      return true;
+      return seen[idx] = true;
   }
 
   bool inst_cv = gutils->isConstantValue(const_cast<Value *>(inst));

Original file line number	Diff line number	Diff line change
`@@ -462,7 +462,7 @@ static inline bool is_value_needed_in_reverse(`
`462`	`462`	`if (auto CI = dyn_cast<CallInst>(inst)) {`
`463`	`463`	`StringRef funcName = getFuncNameFromCall(const_cast<CallInst *>(CI));`
`464`	`464`	`if (funcName == "julia.get_pgcstack" \|\| funcName == "julia.ptls_states")`
`465`		`- return true;`
	`465`	`+ return seen[idx] = true;`
`466`	`466`	`}`
`467`	`467`
`468`	`468`	`bool inst_cv = gutils->isConstantValue(const_cast<Value *>(inst));`