Fix subnull

EnzymeAD · Sep 29, 2024 · b1a2531 · b1a2531
1 parent 680d8bc
commit b1a2531
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/enzyme/Enzyme/GradientUtils.cpp b/enzyme/Enzyme/GradientUtils.cpp
@@ -5743,10 +5743,11 @@ Value *GradientUtils::invertPointerM(Value *const oval, IRBuilder<> &BuilderM,
  Value *ivops[2] = {nullptr, nullptr};
  for (int i = 0; i < 2; i++) {
  auto op = arg->getOperand(i);
+ bool subnull = nullShadow;
  if (!runtimeActivity && !isa<InsertValueInst>(op)) {
-
+ auto vd = TR.query(op);
+ if (!TR.anyFloat(op)) subnull = false;
  if (isConstantValue(op)) {
- auto vd = TR.query(op);
  if (TR.anyPointer(op) && vd[{-1, -1}] != BaseType::Integer) {
  if (!isa<UndefValue>(op) && !isa<ConstantPointerNull>(op)) {
  std::string str;
@@ -5764,7 +5765,7 @@ Value *GradientUtils::invertPointerM(Value *const oval, IRBuilder<> &BuilderM,
  }
  }
  if (!ivops[i]) {
- ivops[i] = invertPointerM(op, bb, nullShadow);
+ ivops[i] = invertPointerM(op, bb, subnull);
  }
  }