P4 to Git Change 1977240 by wchau@wc_hip_vdi on 2019/08/01 16:35:47

SWDEV-180872 - Runtime support changes for Cooperative Group Features
	- Taking into account of SGPRs usage to determine the block size

Affected files ...

... //depot/stg/opencl/drivers/opencl/api/hip/hip_platform.cpp#35 edit


[ROCm/hip commit: 99d54cb7bd]

projects/hip/api/hip/hip_platform.cpp

@@ -492,9 +492,19 @@ hipError_t ihipOccupancyMaxActiveBlocksPerMultiprocessor(int* numBlocks,
 
   // Find threads accupancy per CU => simd_per_cu * GPR usage
   constexpr size_t MaxWavesPerSimd = 8;  // Limited by SPI 32 per CU, hence 8 per SIMD
-  size_t alu_accupancy = device->info().simdPerCU_ *
-    std::min(MaxWavesPerSimd, (wrkGrpInfo->availableVGPRs_ / amd::alignUp(wrkGrpInfo->usedVGPRs_, 4)));
+  size_t VgprWaves = wrkGrpInfo->availableVGPRs_ / amd::alignUp(wrkGrpInfo->usedVGPRs_, 4);
 
+  size_t GprWaves;
+  if (wrkGrpInfo->usedSGPRs_ > 0) {
+    const size_t maxSGPRs = (device->info().gfxipVersion_ < 800) ? 512 : 800;
+    size_t SgprWaves = maxSGPRs / amd::alignUp(wrkGrpInfo->usedSGPRs_, 16);
+    GprWaves = std::min(VgprWaves, SgprWaves);
+  }
+  else {
+    GprWaves = VgprWaves;
+  }
+
+  size_t alu_accupancy = device->info().simdPerCU_ * std::min(MaxWavesPerSimd, GprWaves);
   alu_accupancy *= wrkGrpInfo->wavefrontSize_;
   // Calculate blocks occupancy per CU
   *numBlocks = alu_accupancy / amd::alignUp(blockSize, wrkGrpInfo->wavefrontSize_);