nvshmem_utils.h File Reference

#include "gromacs/gpu_utils/device_stream_manager.h"
#include "gromacs/gpu_utils/devicebuffer_datatype.h"
#include "gromacs/utility/gmxmpi.h"
#include "gromacs/utility/logger.h"

Include dependency graph for nvshmem_utils.h:

This graph shows which files directly or indirectly include this file:

Description

Declarations for NVSHMEM initialization/finalize class. gmxNvshmemHandle takes the MPI communicator and initializes the NVSHMEM over all the ranks involved in the given MPI communicator. This is a collective call for all the ranks in the given MPI comm. After NVSHMEM initialization all NVSHMEM APIs can be safely used.

Author

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