Clear attribute of generated

EnzymeAD · Nov 13, 2024 · 69002c0 · 69002c0
1 parent 624621a
commit 69002c0
Showing 1 changed file with 16 additions and 2 deletions.
diff --git a/enzyme/Enzyme/EnzymeLogic.cpp b/enzyme/Enzyme/EnzymeLogic.cpp
@@ -1696,8 +1696,22 @@ void clearFunctionAttributes(Function *f) {
     if (Arg.hasAttribute(Attribute::StructRet))
       Arg.removeAttr(Attribute::StructRet);
   }
-  if (f->hasFnAttribute(Attribute::OptimizeNone))
-    f->removeFnAttr(Attribute::OptimizeNone);
+
+  Attribute::AttrKind fnattrs[] = {
+#if LLVM_VERSION_MAJOR >= 16
+    Attribute::Memory,
+#endif
+    Attribute::ReadOnly,
+    Attribute::ReadNone,
+    Attribute::WriteOnly,
+    Attribute::WillReturn,
+    Attribute::OptimizeNone
+  };
+  for (auto attr : fnattrs) {
+    if (f->hasFnAttribute(attr)) {
+      f->removeFnAttr(attr);
+    }
+  }
 
   if (f->getAttributes().getRetDereferenceableBytes()) {
     f->removeRetAttr(Attribute::Dereferenceable);