Added Cooperative Groups configure option and header

2024-10-01 13:26:43 -07:00
parent 1baa071edf
commit 63667a3167
5 changed files with 60 additions and 0 deletions
@@ -83,9 +83,21 @@ void SyncTester::launchKernel(dim3 gridSize, dim3 blockSize, int loop,
  roc_shmem_team_split_strided(ROC_SHMEM_TEAM_WORLD, 0, 1, n_pes, nullptr, 0,
                               &team_sync_world_dup);

+#ifdef USE_COOPERATIVE_GROUPS
+  void* kernelParams[] = {(void*)&loop,
+                          (void*)&args.skip,
+                          (void*)&timer,
+                          (void*)&_type,
+                          (void*)&_shmem_context,
+                          (void*)&team_sync_world_dup};
+
+  CHECK_HIP(hipLaunchCooperativeKernel(SyncTest, gridSize, blockSize,
+                                       kernelParams, shared_bytes, stream));
+#else
  hipLaunchKernelGGL(SyncTest, gridSize, blockSize, shared_bytes, stream, loop,
                     args.skip, timer, _type, _shmem_context,
                     team_sync_world_dup);
+#endif /* USE_COOPERATIVE_GROUPS */

  num_msgs = (loop + args.skip) * gridSize.x;
  num_timed_msgs = loop;