nvshmem_utils.cpp File Reference

#include "nvshmem_utils.h"
#include "config.h"
#include "gromacs/gpu_utils/hostallocator.h"
#include "gromacs/utility/basedefinitions.h"
#include "gromacs/utility/exceptions.h"
#include "gromacs/utility/gmxassert.h"

Include dependency graph for nvshmem_utils.cpp:

Description

Definitions 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