Tg blas3 (#1117)

* removing manual blas support

* minimal blas generation

* add blas sanity checks

* emit handle_blas

* emit more parts

* emit caching

* finish rebasing of tblgen-blas

* finish refactor on latest main

* fix fwd-mode

* update tests

* fix Attribute settings

* fix tg for older llvm

* add more const and clarify comments

* add inactive fwd test

* disable dot fallback

* make BLAS fallback optional

* cleanup

* fixup

* cleanup

---------

Co-authored-by: William S. Moses <[email protected]>

Author: ZuseZ4

enzyme/BCLoad/BCLoader.cpp

@@ -16,7 +16,7 @@ using namespace llvm;
 #include "blas_headers.h"
 #undef DATA
 
-bool provideDefinitions(Module &M) {
+bool provideDefinitions(Module &M, std::set<std::string> ignoreFunctions = {}) {
   std::vector<StringRef> todo;
   bool seen32 = false;
   bool seen64 = false;
@@ -27,11 +27,14 @@ bool provideDefinitions(Module &M) {
     int index = 0;
     for (auto postfix : {"", "_", "_64_"}) {
       std::string str;
-      if (strlen(postfix) == 0)
+      if (strlen(postfix) == 0) {
         str = F.getName().str();
-      else if (F.getName().endswith(postfix)) {
-        str = "cblas_" +
-              F.getName().substr(0, F.getName().size() - strlen(postfix)).str();
+        if (ignoreFunctions.count(str)) continue;
+      } else if (F.getName().endswith(postfix)) {
+        auto blasName =
+            F.getName().substr(0, F.getName().size() - strlen(postfix)).str();
+        if (ignoreFunctions.count(blasName)) continue;
+        str = "cblas_" + blasName;
       }
 
       auto found = EnzymeBlasBC.find(str);
@@ -96,8 +99,13 @@ bool provideDefinitions(Module &M) {
 }
 
 extern "C" {
-uint8_t EnzymeBitcodeReplacement(LLVMModuleRef M) {
-  return provideDefinitions(*unwrap(M));
+uint8_t EnzymeBitcodeReplacement(LLVMModuleRef M, char **FncsNamesToIgnore,
+                                 size_t numFncNames) {
+  std::set<std::string> ignoreFunctions = {};
+  for (size_t i = 0; i < numFncNames; i++) {
+    ignoreFunctions.insert(std::string(FncsNamesToIgnore[i]));
+  }
+  return provideDefinitions(*unwrap(M), ignoreFunctions);
 }
 }
 
@@ -107,7 +115,7 @@ class BCLoader final : public ModulePass {
   static char ID;
   BCLoader() : ModulePass(ID) {}
 
-  bool runOnModule(Module &M) override { return provideDefinitions(M); }
+  bool runOnModule(Module &M) override { return provideDefinitions(M, {}); }
 };
 } // namespace