mirror of
https://github.com/NVIDIA/nccl-tests.git
synced 2026-10-26 02:13:31 +00:00
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 <[email protected]>
This commit is contained in:
@@ -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;
|
||||
|
||||
Reference in New Issue
Block a user