From 4817ec47780f4c19b4686eb87221c78ba463228b Mon Sep 17 00:00:00 2001
From: Aryan Salmanpour <aryan.salmanpour@amd.com>
Date: Mon, 28 Sep 2020 15:17:19 -0400
Subject: [PATCH] fix a bug for calculating the prev_sum of MGSyncInfo struct
 used in multi-gpu cooperative launch

SWDEV-251619

Change-Id: I6566bfc9384bb4677562085991a7a01d3cea3935
---
 hipamd/rocclr/hip_module.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/hipamd/rocclr/hip_module.cpp b/hipamd/rocclr/hip_module.cpp
index 4a09cc6ed0..b72ee1a5a2 100755
--- a/hipamd/rocclr/hip_module.cpp
+++ b/hipamd/rocclr/hip_module.cpp
@@ -537,7 +537,7 @@ hipError_t ihipLaunchCooperativeKernelMultiDevice(hipLaunchParams* launchParamsL
     if (result != hipSuccess) {
       break;
     }
-    prevGridSize += launch.gridDim.x * launch.gridDim.y * launch.gridDim.z;
+    prevGridSize += globalWorkSizeX * globalWorkSizeY * globalWorkSizeZ;
   }
 
   // Sync the execution streams on all devices