extra Blas header to include instead of inc files

EnzymeAD · Oct 24, 2023 · d57adb3 · d57adb3
1 parent 2d7611a
commit d57adb3
Show file tree

Hide file tree

Showing 4 changed files with 10 additions and 4 deletions.
diff --git a/enzyme/Enzyme/AdjointGenerator.h b/enzyme/Enzyme/AdjointGenerator.h
@@ -3748,8 +3748,7 @@ class AdjointGenerator
   }
 
 // first one allows adding attributes to blas functions declared in the second
-#include "BlasAttributor.inc"
-#include "BlasDerivatives.inc"
+#include "BlasHeader.h"
 
   void visitOMPCall(llvm::CallInst &call) {
     using namespace llvm;

diff --git a/enzyme/Enzyme/BlasHeader.h b/enzyme/Enzyme/BlasHeader.h
@@ -0,0 +1,7 @@
+// Adds readonly/writeonly/... attributes to a BLAS function
+void attributeBLAS(BlasInfo blas, llvm::Function *F);
+// Adds attributes to a few more non blas functions
+void attributeTablegen(llvm::Function &F);
+
+// Differentiates BLAS functions
+bool handleBLAS(llvm::CallInst &call, llvm::Function *called,BlasInfo blas,const std::vector<bool> &overwritten_args);
diff --git a/enzyme/Enzyme/Enzyme.cpp b/enzyme/Enzyme/Enzyme.cpp
@@ -90,7 +90,7 @@
 #include "llvm/Transforms/Utils/Mem2Reg.h"
 #endif
 
-#include "BlasAttributor.inc"
+#include "BlasHeader.h"
 
 #include "CApi.h"
 using namespace llvm;

diff --git a/enzyme/Enzyme/PreserveNVVM.cpp b/enzyme/Enzyme/PreserveNVVM.cpp
@@ -299,7 +299,7 @@ bool preserveLinkage(bool Begin, Function &F) {
   return false;
 }
 
-void attributeBLAS(BlasInfo blas, llvm::Function *F);
+#include "BlasHeader.h"
 
 bool preserveNVVM(bool Begin, Function &F) {
   bool changed = false;