Fix weird allocation (#1918)

EnzymeAD · Jun 5, 2024 · 5325724 · 5325724
1 parent 46647d1
commit 5325724
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/enzyme/Enzyme/Utils.cpp b/enzyme/Enzyme/Utils.cpp
@@ -301,6 +301,10 @@ Value *CreateAllocation(IRBuilder<> &Builder, llvm::Type *T, Value *Count,
     res = unwrap(CustomAllocator(wrap(&Builder), wrap(T), wrap(Count),
                                  wrap(Align), isDefault,
                                  ZeroMem ? &wzeromem : nullptr));
+    if (isa<UndefValue>(res))
+      return res;
+    if (isa<Constant>(res))
+      return res;
     if (auto I = dyn_cast<Instruction>(res))
       I->setName(Name);