AI/rocm-systems
2
0 Forc 0
Cód Saincheisteanna Iarratais Tarraingthe Gníomhartha Pacáistí Tionscadail Scaoileann Vicí Gníomhaíocht

Fix thread index calculation in __hip_malloc

Brabhsáil Foinse 
[ROCm/hip commit: 03320890de]
Tá an tiomantas seo le fáil i:
Yaxun Sam Liu tiomanta 2018-07-27 16:24:51 -04:00
1 tuismitheoir 289bfe5f95
tiomantas da2fd69c44
2 comhad athraithe
+24 -6
Íoslódáil an comhad paiste Íoslódáil Comhad Diff Leathnaigh gach comhaid Laghdaigh gach comhaid

Níor tháinig aon chomhaid le do chuardach

projects/hip/include/hip/hcc_detail/hip_memory.h
+4 -1
Féach ar an gComhad
@@ -51,7 +51,10 @@ extern "C" inline __device__ void* __hip_malloc(size_t size) {
uint32_t totalThreads =
hipBlockDim_x * hipGridDim_x * hipBlockDim_y
* hipGridDim_y * hipBlockDim_z * hipGridDim_z;
uint32_t currentWorkItem = hipThreadIdx_x + hipBlockDim_x * hipBlockIdx_x;
uint32_t currentWorkItem = hipThreadIdx_x + hipBlockDim_x * hipBlockIdx_x
+ (hipThreadIdx_y + hipBlockDim_y * hipBlockIdx_y) * hipBlockDim_x
+ (hipThreadIdx_z + hipBlockDim_z * hipBlockIdx_z) * hipBlockDim_x
* hipBlockDim_y;
uint32_t numHeapsPerWorkItem = __HIP_NUM_PAGES / totalThreads;
uint32_t heapSizePerWorkItem = __HIP_SIZE_OF_HEAP / totalThreads;