maxjhandsome
/
llvm-project
forked from OSchip/llvm-project
0
0
Fork 0
Code Issues Pull Requests Packages Releases Wiki Activity

[VENTUS][#119]Fix workgroup function barrier scope

This commit is contained in:
ziliangzl 2024-05-14 14:03:20 +08:00
parent d977b0bf8b
commit 6a8e4d4667
1 changed files with 9 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
libclc/riscv32/lib/workgroup/wgreduce.cl
View File

@ -16,16 +16,16 @@ work_group_reduce_add(TYPE a) \
\
p[i] = a; \
\
work_group_barrier(CLK_LOCAL_MEM_FENCE); \
work_group_barrier(CLK_GLOBAL_MEM_FENCE); \
if (i == 0) { \
TYPE res = 0; \
for (int j = 0; j < n; j++) \
res += p[j]; \
p[0] = res; \
} \
work_group_barrier(CLK_LOCAL_MEM_FENCE); \
work_group_barrier(CLK_GLOBAL_MEM_FENCE); \
a = p[0]; \
work_group_barrier(CLK_LOCAL_MEM_FENCE); \
work_group_barrier(CLK_GLOBAL_MEM_FENCE); \
return a; \
} \
\
@ -42,16 +42,16 @@ work_group_reduce_max(TYPE a) \
\
p[i] = a; \
\
work_group_barrier(CLK_LOCAL_MEM_FENCE); \
work_group_barrier(CLK_GLOBAL_MEM_FENCE); \
if (i == 0) { \
TYPE res = p[0]; \
for (int j = 0; j < n; j++) \
res = p[j] > res ? p[j] : res; \
p[0] = res; \
} \
work_group_barrier(CLK_LOCAL_MEM_FENCE); \
work_group_barrier(CLK_GLOBAL_MEM_FENCE); \
a = p[0]; \
work_group_barrier(CLK_LOCAL_MEM_FENCE); \
work_group_barrier(CLK_GLOBAL_MEM_FENCE); \
return a; \
} \
\
@ -68,16 +68,16 @@ work_group_reduce_min(TYPE a) \
\
p[i] = a; \
\
work_group_barrier(CLK_LOCAL_MEM_FENCE); \
work_group_barrier(CLK_GLOBAL_MEM_FENCE); \
if (i == 0) { \
TYPE res = p[0]; \
for (int j = 0; j < n; j++) \
res = p[j] < res ? p[j] : res; \
p[0] = res; \
} \
work_group_barrier(CLK_LOCAL_MEM_FENCE); \
work_group_barrier(CLK_GLOBAL_MEM_FENCE); \
a = p[0]; \
work_group_barrier(CLK_LOCAL_MEM_FENCE); \
work_group_barrier(CLK_GLOBAL_MEM_FENCE); \
return a; \
}