summaryrefslogtreecommitdiff
path: root/libclc/amdgcn/lib/workitem/get_local_size.cl
blob: c398b7eb5a9d1fd0bbf20d1278db046a13bb5c1b (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
#include <clc/clc.h>

uint __clc_amdgcn_get_local_size_x(void) __asm("llvm.r600.read.local.size.x");
uint __clc_amdgcn_get_local_size_y(void) __asm("llvm.r600.read.local.size.y");
uint __clc_amdgcn_get_local_size_z(void) __asm("llvm.r600.read.local.size.z");

_CLC_DEF _CLC_OVERLOAD size_t get_local_size(uint dim) {
  switch (dim) {
  case 0:
    return __clc_amdgcn_get_local_size_x();
  case 1:
    return __clc_amdgcn_get_local_size_y();
  case 2:
    return __clc_amdgcn_get_local_size_z();
  default:
    return 1;
  }
}