From 26be02998adfa90a0c06c5effee4c3de4faba370 Mon Sep 17 00:00:00 2001
From: Ahsan Pervaiz <apervaiz@nvidia.com>
Date: Tue, 3 Mar 2026 14:05:52 -0800
Subject: [PATCH] Request GIN to be explicitly enabled in all to all test

Based on the changes in NCCL v2.29.3, update the alltoall test to
either provide a ginConnectionType or set ginForceEnable to true.

Signed-off-by: Ahsan Pervaiz <apervaiz@nvidia.com>
---
 src/alltoall.cu | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/src/alltoall.cu b/src/alltoall.cu
index d0289a1..815b679 100644
--- a/src/alltoall.cu
+++ b/src/alltoall.cu
@@ -80,6 +80,11 @@ testResult_t AlltoAllGetDevCommRequirements(int deviceImpl, ncclDevCommRequireme
       }
       reqs->barrierCount = deviceCtaCount;
       reqs->ginSignalCount = deviceCtaCount;
+#if NCCL_VERSION_CODE >= NCCL_VERSION(2, 29, 7)
+      reqs->ginConnectionType = NCCL_GIN_CONNECTION_FULL
+#else
+      reqs->ginForceEnable = true;
+#endif
       return testSuccess;
     default:
       return testNotImplemented;