What UCCL-Tran Changes for Collectives and ML Applications

UCCL-Tran's value is not measured by one abstract transport benchmark. It appears when a particular topology and workload expose a bottleneck that software control can address. This chapter reads the collective and application results in that context, then follows the cost of using the system: extra host CPU, driver and GPUDirect prerequisites, and different implementations for different NICs. The aim is to identify when a reported speedup is evidence of a transferable mechanism and when it is mainly a property of the testbed.

The paper evaluates UCCL-Tran across NVIDIA ConnectX-7, Broadcom Thor-2, and AWS EFA environments, using both NVIDIA and AMD collective libraries. Those comparisons are valuable precisely because they are not identical. A cross-rack Ethernet experiment stresses path collisions; a rail-optimized topology changes the congestion problem; EFA exposes a different hardware control boundary; and a same-rack InfiniBand experiment asks whether software can keep up when the network itself is not the main obstacle.

The biggest gains answer specific bottlenecks

The evaluation uses four environments. The first, called CX_ETH in the paper, has six servers connected across Ethernet racks, with eight 400-Gbps NVIDIA ConnectX-7 NICs and eight H100 GPUs per server. The second has four servers with eight Broadcom Thor-2 400-Gbps NICs and eight AMD MI300X GPUs, using a rail-optimized Ethernet topology. The third is an AWS p4d-style EFA environment with four 100-Gbps EFA NICs and eight A100 GPUs per server. The fourth is a two-server CX_IB setup with eight ConnectX-7 NICs and eight H100 GPUs in the same rack.

These environments turn the same question into four different experiments: can UCCL-Tran avoid a bad route, can it control all-to-all traffic on a topology already designed for it, can host CPU software outperform a constrained NIC-side processor, and can it match a hardware transport when path collisions are largely absent? The collective tests use NCCL or RCCL and vary message sizes from 1 MB to 1 GB. UCCL-Tran generally uses 256 paths, while the hardware baselines use their normal or recommended queue-pair settings.

flowchart LR A["Traffic and topology"] --> B["UCCL-Tran mechanism"] B --> C["Collective bandwidth or latency"] C --> D["Training or serving outcome"] B --> E["CPU, driver, and memory cost"] A --> F["Evidence qualification"] F -. "conditions matter" .-> C

On CX_ETH, the baseline ConnectX-7 transport suffers from flow collisions in the cross-rack fat-tree. As message sizes increase, the hardware's congestion response can back off sharply even though the network has other paths. UCCL-Tran's software path selection and congestion control improve allreduce by up to 2.32, 1.60, or 1.24 times when the ConnectX-7 baseline uses 4, 8, or 16 QPs per connection. For all-to-all, the corresponding improvements reach 1.79, 3.82, and 4.54 times. The exact multiplier depends on the baseline configuration, but the causal pattern is stable: the advantage is largest when coordinated flows are colliding and software can select around the congestion.