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

EnzymeAD · Jul 16, 2022 · 8b8cd30 · 8b8cd30
1 parent 969ddce
commit 8b8cd30
Showing 1 changed file with 3 additions and 2 deletions.
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);