SWDEV-220503: this_grid().thread_rank() gives incorrect result (#1808)

* fix a minor bug while computing this.grid()::thread_rank()

[ROCm/clr commit: 0605426049]
Этот коммит содержится в:
mshivama
2020-01-24 16:23:28 +05:30
коммит произвёл Maneesh Gupta
родитель fec5105d9c
Коммит a49cbb1580
+2 -2
Просмотреть файл
@@ -106,7 +106,7 @@ __CG_STATIC_QUALIFIER__ uint32_t size() {
}
__CG_STATIC_QUALIFIER__ uint32_t thread_rank() {
// Compute global id of the workgroup to which the current threads belongs to
// Compute global id of the workgroup to which the current thread belongs to
uint32_t blkIdx =
(uint32_t)((hipBlockIdx_z * hipGridDim_y * hipGridDim_x) +
(hipBlockIdx_y * hipGridDim_x) +
@@ -115,7 +115,7 @@ __CG_STATIC_QUALIFIER__ uint32_t thread_rank() {
// Compute total number of threads being passed to reach current workgroup
// within grid
uint32_t num_threads_till_current_workgroup =
(uint32_t)(blkIdx * (hipBlockIdx_x * hipBlockIdx_y * hipBlockIdx_z));
(uint32_t)(blkIdx * (hipBlockDim_x * hipBlockDim_y * hipBlockDim_z));
// Compute thread local rank within current workgroup
uint32_t local_thread_rank =