offload/unittests/OffloadAPI/device_code/localmem_static.cpp


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17

#include <gpuintrin.h>
#include <stdint.h>

[[clang::loader_uninitialized]]
__gpu_local uint32_t shared_mem[64];

extern "C" __gpu_kernel void localmem_static(uint32_t *out) {
  shared_mem[__gpu_thread_id(0)] = 2;

  __gpu_sync_threads();

  if (__gpu_thread_id(0) == 0) {
    out[__gpu_block_id(0)] = 0;
    for (uint32_t i = 0; i < __gpu_num_threads(0); i++)
      out[__gpu_block_id(0)] += shared_mem[i];
  }
}