OSchip
/
llvm-project
mirror of https://github.com/THU-DSP-LAB/llvm-project.git
13
0
Fork 2
Code Issues Packages Projects Releases Wiki Activity

[VENTUS][RISCV][fix] Fix get_global_id_z function bug

This commit is contained in:
zhoujing 2023-07-19 19:32:26 +08:00
parent 1a6ead3f43
commit 258d76a7c8
1 changed files with 3 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
libclc/riscv32/lib/workitem/workitem.S
View File

@ -236,12 +236,9 @@ __builtin_riscv_global_id_z:
call __builtin_riscv_workitem_id_z call __builtin_riscv_workitem_id_z
csrr a0, CSR_KNL # Get kernel metadata buffer csrr a0, CSR_KNL # Get kernel metadata buffer
csrr t1, CSR_GID_Z # Get group_id_z csrr t1, CSR_GID_Z # Get group_id_z
csrr t2, CSR_TID lw t2, KNL_LC_SIZE_Z(a0) # Get local_size_z
vid.v v2 mul t2, t2, t1 # CSR_GID_Z * local_size_z
vadd.vx v2, v2, t2 # workitem_id_z vadd.vx v0, v0, t2 # global_id_z
lw t3, KNL_LC_SIZE_Z(a0) # Get local_size_z
mul t3, t1, t3 # CSR_GID_Z * local_size_z
vadd.vv v0, v2, v0 # global_id_z
lw ra, -4(sp) lw ra, -4(sp)
addi sp, sp, -4 addi sp, sp, -4
ret ret