Function block_id_in_cluster

include/common.h:435–441

Returns the relative dim3 block rank local to the cluster.

// Returns the relative dim3 block rank local to the cluster.
DEVICE dim3 block_id_in_cluster() {
  uint32_t x, y, z;
  asm volatile("mov.u32 %0, %cluster_ctaid.x;\n" : "=r"(x) :);
  asm volatile("mov.u32 %0, %cluster_ctaid.y;\n" : "=r"(y) :);
  asm volatile("mov.u32 %0, %cluster_ctaid.z;\n" : "=r"(z) :);
  return {x, y, z};
}

// Returns the warp id in block
DEVICE int warp_id_in_block() { return threadIdx.x / WARP_SIZE; }