Gromacs  2026.0-dev-20250711-6857db4
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Groups Pages
List of all members | Public Member Functions
gmx::NvshmemManager Class Reference

#include <gromacs/gpu_utils/nvshmem_manager.h>

Description

Duplicates the communicator and initializes NVSHMEM over it. This is a collective call for all the ranks in the given MPI comm. After NVSHMEM initialization all NVSHMEM APIs can be safely used.

May issue warnings advisory notes about the runtime environment for NVSHMEM use.

Frees the communicator in the destructor after finalizing the NVSHMEM library.

Author
Mahesh Doijade mdoij.nosp@m.ade@.nosp@m.nvidi.nosp@m.a.co.nosp@m.m

Public Member Functions

 NvshmemManager (const MDLogger &mdlog, MPI_Comm comm)
 

The documentation for this class was generated from the following files: