Help us improve your experience.

Let us know what you think.

Do you have time for a two-minute survey?

 
 

Appendix A – How to Run a NCCL Test Using Autoconfigured IPv6 Address

To run a model or NCCL test using a global IPv6 addresses assigned either statically or automatically via SLAAC the value of the NCCL_IB_GID_INDEX variable must be adjusted. The NCCL_IB_GID_INDEX variable defines the Global ID index used by RoCE (RDMA) communication.

The default value is -1, which means that NCCL will automatically select the correct GID index based on the active link layer of the InfiniBand device. If the link layer is Ethernet (RoCE), NCCL will use the GID index that returns a GID with RoCE v2 support (usually GID index 3, depending on driver/firmware).

For more details you can review Nvidia’s Environment Variables — NCCL 2.27.3 documentation

To find the GID for the desired address, use the following command:

To find the mellanox interface name you can use the following script:

Example:

Once you have identified the GID you can run a NCCL test as shown in the example:

The following script provides mapping information between the Mellanox interface names, the NIC numbers, and the user assigned interface names (e.g. gpu0_eth). It also provides mapping information between the interfaces and the GPUs.

Example: