From bc519fd7334272246536e03cf294bd8551176806 Mon Sep 17 00:00:00 2001
From: akolliasAMD <99202231+akolliasAMD@users.noreply.github.com>
Date: Tue, 8 Oct 2024 08:43:29 -0600
Subject: [PATCH] disabled wbinvl1 for gfx9x on ll128 (#1365)

---
 src/device/prims_ll128.h | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/device/prims_ll128.h b/src/device/prims_ll128.h
index 2b9cefa505..8b72b35780 100644
--- a/src/device/prims_ll128.h
+++ b/src/device/prims_ll128.h
@@ -13,7 +13,7 @@
 #define NCCL_LL128_FLAGTHREAD (NCCL_LL128_LINEELEMS-1)
 
 #ifndef RCCL_USE_WBINVL1_VOL
-#if defined(__GFX8__) || defined(__GFX9__)
+#if defined(__GFX8__) || defined(__gfx906__) || defined(__gfx908__) || defined(__gfx90a__)
 #define RCCL_USE_WBINVL1_VOL 1
 #else
 #define RCCL_USE_WBINVL1_VOL 0