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;