TheRock/rocm-systems/projects/rccl/examples/01_communicators

NCCL Communicator Examples

Overview

This directory contains minimal examples that demonstrate NCCL communicator lifecycle management (creation, query, and destruction) using different initialization patterns.

Examples

01_multiple_devices_single_process

Multiple Devices Single Process

• Pattern: Single process manages all GPUs
• API: ncclCommInitAll (no external coordination)
• Use case: Simple single-node applications
• Key features:
  • Simplest initialization method
  • No MPI or threading required
  • Automatic rank assignment (0 to n-1)
  • Cannot span multiple nodes

Run command:

./01_multiple_devices_single_process/multiple_devices_single_process

02_one_device_per_pthread

One Device per Thread with pthreads

• Pattern: One thread per GPU within single process
• API: ncclCommInitRank with pthread coordination
• Use case: Single-node multi-GPU, thread-based parallelism
• Key features:
  • pthread barriers for synchronization
  • Shared memory for unique ID
  • Lower overhead than multi-process
  • Cannot span multiple nodes

Run command:

[NTHREADS=n] ./02_one_device_per_pthread/one_device_per_pthread

03_one_device_per_process_mpi

One Device per Process with MPI

• Pattern: One MPI process per GPU
• API: ncclCommInitRank with MPI coordination
• Use case: Multi-node clusters, distributed training
• Key features:
  • MPI broadcast for unique ID distribution
  • Process-to-GPU mapping by local MPI ranks
  • Scalable to multiple nodes

Run command:

mpirun -np <num_processes> ./03_one_device_per_process_mpi/one_device_per_process_mpi

Choosing the Right Approach

Feature	ncclCommInitAll	pthread	MPI
Multi-node support	✗	✗	✓
Process isolation	✗	✗	✓
Setup complexity	Low	Medium	High
Memory overhead	Low	Medium	High
Best for	Simple test	Single-node apps	Clusters

When to use each:

• ncclCommInitAll: Development, testing, simple single-node apps
• pthread: Single-node with thread-based parallelism needs
• MPI: Production distributed training, multi-node setups

Building

Quick Start

# Build all examples [or single directory]
make [directory]

# Test all examples
make test

Individual Examples

# Build specific example
make 01_multiple_devices_single_process
make 02_one_device_per_pthread
make 03_one_device_per_process_mpi

# Test individual example
cd 01_multiple_devices_single_process && make test
cd 02_one_device_per_pthread && make test
cd 03_one_device_per_process_mpi && make test

References

• NVIDIA NCCL User Guide Examples
• NCCL API Reference
• HIP Programming Guide
• MPI Standard