Mark getenv/strtol/fwrite as nondifferentiable (#1868)

* Mark getenv/strtol/fwrite as nondifferentiable * also mark no escaping alloc * add nlopt
EnzymeAD · May 8, 2024 · f22916f · f22916f
1 parent f3dd860
commit f22916f
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 2 deletions.
diff --git a/enzyme/Enzyme/ActivityAnalysis.cpp b/enzyme/Enzyme/ActivityAnalysis.cpp
@@ -209,6 +209,9 @@ const StringSet<> KnownInactiveFunctions = {
     "__cxa_guard_acquire",
     "__cxa_guard_release",
     "__cxa_guard_abort",
+    "getenv",
+    "strtol",
+    "fwrite",
     "snprintf",
     "sprintf",
     "printf",

diff --git a/enzyme/Enzyme/Enzyme.cpp b/enzyme/Enzyme/Enzyme.cpp
@@ -343,7 +343,8 @@ bool attributeKnownFunctions(llvm::Function &F) {
                      "createERmm" ||
       F.getName() ==
           "_ZNKSt8__detail20_Prime_rehash_policy14_M_need_rehashEmmm" ||
-      F.getName() == "fprintf") {
+      F.getName() == "fprintf" || F.getName() == "fwrite" ||
+      F.getName() == "strtol" || F.getName() == "getenv") {
     changed = true;
     F.addAttribute(
         AttributeList::FunctionIndex,

diff --git a/enzyme/Enzyme/EnzymeLogic.cpp b/enzyme/Enzyme/EnzymeLogic.cpp
@@ -6377,7 +6377,9 @@ llvm::Function *EnzymeLogic::CreateNoFree(RequestContext context, Function *F) {
                          "MPI_Allreduce",
                          "lgamma",
                          "lgamma_r",
-                         "__kmpc_global_thread_num"};
+                         "__kmpc_global_thread_num",
+                         "nlopt_force_stop"
+  };
   // clang-format on
 
   if (startsWith(F->getName(), "_ZNSolsE") || NoFrees.count(F->getName()))