[Openmp-commits] [openmp] [OpenMP] Basic BumpAllocator for (AMD)GPUs (PR #69806)
Joseph Huber via Openmp-commits
openmp-commits at lists.llvm.org
Fri Oct 20 19:55:35 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 Openmp-commits
mailing list