[DirectX] Implement the resource.store.rawbuffer intrinsic (#121282)

[llvm-project.git] / libclc / ptx-nvidiacl / lib / workitem / get_num_groups.cl

#include <clc/clc.h>

_CLC_DEF _CLC_OVERLOAD size_t get_num_groups(uint dim) {
  switch (dim) {
  case 0:  return __nvvm_read_ptx_sreg_nctaid_x();
  case 1:  return __nvvm_read_ptx_sreg_nctaid_y();
  case 2:  return __nvvm_read_ptx_sreg_nctaid_z();
  default: return 0;
  }
}