Add check for ROCM version for syncwarp (#220)

* Add check for ROCM version for syncwarp

Add check for ROCM version for syncwarp (#220)
* Add check for ROCM version for syncwarp
ab945c1d · David Sidler · GitHub · a7dfea3d · ab945c1d
Unverified Commit ab945c1d authored Dec 05, 2025 by David Sidler Committed by GitHub Dec 05, 2025
Show whitespace changes
Inline Side-by-side

Showing with 5 additions and 0 deletions

src/header/TransferBench.hpp src/header/TransferBench.hpp +5 -0

No files found.
--- a/src/header/TransferBench.hpp
+++ b/src/header/TransferBench.hpp
@@ -3173,7 +3173,12 @@ static bool IsConfiguredGid(union ibv_gid const& gid)
    // Wait for all threads to finish
    if (seType == 1) {
      // For warp-level, sync within warp only
+ #if defined(__HIP_PLATFORM_AMD__) && (HIP_VERSION_MAJOR < 7)
+      __builtin_amdgcn_wave_barrier();
+ #else
      __syncwarp();
+ #endif
    } else {
      // For threadblock-level, sync all threads
      __syncthreads();