[clang] [OpenMP] Basic BumpAllocator for (AMD)GPUs (PR #69806)

Joseph Huber via cfe-commits cfe-commits at lists.llvm.org
Fri Oct 20 19:55:37 PDT 2023


================
@@ -36,6 +46,15 @@ extern "C" {
 #pragma omp end declare variant
 
 #ifdef __AMDGCN__
+#pragma omp begin declare variant match(                                       \
+        device = {arch(amdgcn)},                                               \
+            implementation = {extension(disable_implicit_base)})
+
+void *malloc(size_t Size) { return llvm_device_malloc(Size); }
+void free(void *Ptr) { llvm_device_free(Ptr); }
----------------
jhuber6 wrote:

Unsure how to feel about this, here we always replace `malloc` calls with the one inside the OpenMP device runtime library won't it?

https://github.com/llvm/llvm-project/pull/69806


More information about the cfe-commits mailing list