Skip to content

Add NCCL Jacobi example #1576

Description

@nv-snidhan

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

Metadata

Metadata

Assignees

Labels

docsImprovements or additions to documentation

Fields

No fields configured for Documentation.

Projects

No projects

Milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions