a3f423cf57
This patch adds the `llvm_omp_target_dynamic_shared_alloc` function to the `omp.h` header file so users can access it by default. Also changed the name to keep it consistent with the other target allocators. Added some documentation so users know how to use it. Didn't add the interface for Fortran since there's no way to test it right now. Reviewed By: jdoerfert Differential Revision: https://reviews.llvm.org/D123246
26 lines
630 B
C
26 lines
630 B
C
// RUN: %libomptarget-compile-nvptx64-nvidia-cuda
|
|
// RUN: env LIBOMPTARGET_SHARED_MEMORY_SIZE=256 \
|
|
// RUN: %libomptarget-run-nvptx64-nvidia-cuda | %fcheck-nvptx64-nvidia-cuda
|
|
// REQUIRES: nvptx64-nvidia-cuda
|
|
|
|
#include <omp.h>
|
|
#include <stdio.h>
|
|
|
|
int main() {
|
|
int x;
|
|
#pragma omp target parallel map(from : x)
|
|
{
|
|
int *buf = llvm_omp_target_dynamic_shared_alloc() + 252;
|
|
#pragma omp barrier
|
|
if (omp_get_thread_num() == 0)
|
|
*buf = 1;
|
|
#pragma omp barrier
|
|
if (omp_get_thread_num() == 1)
|
|
x = *buf;
|
|
}
|
|
|
|
// CHECK: PASS
|
|
if (x == 1 && llvm_omp_target_dynamic_shared_alloc() == NULL)
|
|
printf("PASS\n");
|
|
}
|