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

extern "C" __gpu_kernel void gridsize(uint32_t *nblocks, uint32_t *nthreads) {
  if (__gpu_block_id(0) == 0 && __gpu_block_id(1) == 0 &&
      __gpu_block_id(2) == 0 && __gpu_thread_id(0) == 0 &&
      __gpu_thread_id(1) == 0 && __gpu_thread_id(2) == 0) {
    for (int i = 0; i < 3; ++i) {
      nblocks[i] = __gpu_num_blocks(i);
      nthreads[i] = __gpu_num_threads(i);
    }
  }
}