gmx::GpuHaloExchangeNvshmemHelper Class Reference

#include <gromacs/domdec/gpuhaloexchange.h>

Description

Handles NVSHMEM aspects of GPU Halo exchange.

GPU halo exchange requires extra signal buffers with NVSHMEM, as well as the ability to coordinate with a possible PME-only rank to arrange for global communication and symmetric allocations.

Public Member Functions
	GpuHaloExchangeNvshmemHelper (const gmx_domdec_t &dd, const DeviceContext &context, const DeviceStream &stream, const std::optional< int > &peerRank)
void	reinit ()
	Re-initialize after domain repartitioning.
DeviceBuffer< uint64_t >	getSyncBuffer () const
	Return the sync buffer.
int	totalPulsesAndDims () const
	Return the total number of DD pulses and dimensions.
void	freeHaloExchangeBuffers ()
	Permit symmetric deallocation.

---

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

• src/gromacs/domdec/gpuhaloexchange.h
• src/gromacs/domdec/gpuhaloexchange_impl.cpp
• src/gromacs/domdec/gpuhaloexchange_impl_gpu.cpp