Fix builtin macros for memory operations

vgvassilev · Feb 26, 2024 · d78b389 · d78b389
1 parent 7e85226
commit d78b389
Showing 1 changed file with 18 additions and 3 deletions.
diff --git a/lib/Differentiator/CladUtils.cpp b/lib/Differentiator/CladUtils.cpp
@@ -8,6 +8,7 @@
 #include "clang/AST/RecursiveASTVisitor.h"
 #include "clang/Basic/SourceLocation.h"
 #include "clang/Sema/Lookup.h"
+#include "clang/Basic/Builtins.h"
 #include "llvm/ADT/SmallVector.h"
 #include "clad/Differentiator/Compatibility.h"
 
@@ -643,17 +644,31 @@ namespace clad {
     }
 
     bool IsMemoryAllocationFunction(const clang::FunctionDecl* FD) {
+
+      #if CLANG_VERSION_MAJOR > 12
       if (FD->getBuiltinID() == Builtin::BImalloc)
         return true;
-      if (FD->getBuiltinID() == Builtin::BIcalloc)
+      if (FD->getBuiltinID() == Builtin::ID::BIcalloc)
+        return true;
+      if (FD->getBuiltinID() == Builtin::ID::BIrealloc)
+        return true;
+      #else
+      if (FD->getNameAsString() == "malloc")
+        return true;
+      if (FD->getNameAsString() == "calloc")
         return true;
-      if (FD->getBuiltinID() == Builtin::BIrealloc)
+      if (FD->getNameAsString() == "realloc")
         return true;
+      #endif
       return false;
     }
 
     bool IsMemoryDeallocationFunction(const clang::FunctionDecl* FD) {
-      return FD->getBuiltinID() == Builtin::BIfree;
+      #if CLANG_VERSION_MAJOR > 12
+      return FD->getBuiltinID() == Builtin::ID::BIfree;
+      #else
+      return FD->getNameAsString() == "free";
+      #endif
     }
   } // namespace utils
 } // namespace clad