TL/UCP: Force ring allgather for Grace

openucx · Feb 24, 2025 · bc2fecc · bc2fecc
1 parent bc996dd
commit bc2fecc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/contrib/ucc.conf b/contrib/ucc.conf
@@ -100,7 +100,7 @@ UCC_TL_UCP_TUNE=allreduce:0-4k:@0#allreduce:4k-inf:@1
 [vendor=nvidia model=grace team_size=144 sock=72 nnodes=1]
 UCC_TL_UCP_ALLREDUCE_KN_RADIX=0-4k:host:2
 UCC_TL_UCP_ALLREDUCE_SRA_KN_RADIX=4096-8192:host:96,8192-16384:host:4,16384-32768:host:6,32768-65536:host:18,65536-131072:host:32,131072-262144:host:72,262144-524288:host:3,524288-1048576:host:2
-UCC_TL_UCP_TUNE=allreduce:0-4k:@0#allreduce:4k-inf:@1
+UCC_TL_UCP_TUNE=allreduce:0-4k:@0#allreduce:4k-inf:@1#allgather:37748736-75497472:host:@1
 
 [vendor=nvidia model=grace team_size=128 sock=64 nnodes=1]
 UCC_TL_UCP_ALLREDUCE_KN_RADIX=0-4k:host:2