Fix wrapped value used in chain rule (rust-lang#732)

tgymnich · web-flow · commit 8b8cd30c5d9c · 2022-07-16T15:44:33.000-04:00
diff --git a/enzyme/Enzyme/GradientUtils.cpp b/enzyme/Enzyme/GradientUtils.cpp
@@ -4248,8 +4248,9 @@ Value *GradientUtils::invertPointerM(Value *const oval, IRBuilder<> &BuilderM,
         }
       }
       if (auto C = dyn_cast<Constant>(ip)) {
-        auto rule = [&](Value *ip) {
-          return ConstantExpr::getCast(arg->getOpcode(), C, arg->getType());
+        auto rule = [&arg](Value *ip) {
+          return ConstantExpr::getCast(arg->getOpcode(), cast<Constant>(ip),
+                                       arg->getType());
         };
 
         return applyChainRule(arg->getType(), bb, rule, ip);

Original file line number	Diff line number	Diff line change
`@@ -4248,8 +4248,9 @@ Value GradientUtils::invertPointerM(Value const oval, IRBuilder<> &BuilderM,`
`4248`	`4248`	`}`
`4249`	`4249`	`}`
`4250`	`4250`	`if (auto C = dyn_cast<Constant>(ip)) {`
`4251`		`- auto rule = [&](Value *ip) {`
`4252`		`- return ConstantExpr::getCast(arg->getOpcode(), C, arg->getType());`
	`4251`	`+ auto rule = [&arg](Value *ip) {`
	`4252`	`+ return ConstantExpr::getCast(arg->getOpcode(), cast<Constant>(ip),`
	`4253`	`+ arg->getType());`
`4253`	`4254`	`};`
`4254`	`4255`
`4255`	`4256`	`return applyChainRule(arg->getType(), bb, rule, ip);`