19 lines · cpp
1#include <gpuintrin.h>2#include <stdint.h>3 4extern "C" {5 6[[gnu::visibility("default")]]7uint32_t global[64];8 9__gpu_kernel void write() {10 global[__gpu_thread_id(0)] = __gpu_thread_id(0);11 global[__gpu_thread_id(0)] *= 2;12}13 14__gpu_kernel void read(uint32_t *out) {15 out[__gpu_thread_id(0) + (__gpu_num_threads(0) * __gpu_block_id(0))] =16 global[__gpu_thread_id(0)];17}18} // extern "C"19