Description
Add an example that uses NCCL (via nccl4py) with Warp arrays for multi-GPU halo exchange: a distributed Jacobi solver for the Laplace equation, the NCCL counterpart to the mpi4py example from GH-475.
Context
Gives users a multi-GPU reference to compare against a single-GPU baseline, mirroring the NCCL variants in NVIDIA/multi-gpu-programming-models
Description
Add an example that uses NCCL (via
nccl4py) with Warp arrays for multi-GPU halo exchange: a distributed Jacobi solver for the Laplace equation, the NCCL counterpart to thempi4pyexample from GH-475.Context
Gives users a multi-GPU reference to compare against a single-GPU baseline, mirroring the NCCL variants in NVIDIA/multi-gpu-programming-models