Module GPU Utils

Collaboration diagram for Module GPU Utils:

Description

A brief description for Module GPU Utils.

Classes
class	gmx::NvshmemManager
	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. More...

Files
file	device_context.cpp
	Implements the DeviceContext for CPU build.
file	device_context.h
	Declarations for DeviceContext class.
file	device_context_ocl.cpp
	Implements the DeviceContext for OpenCL.
file	device_context_sycl.cpp
	Implements the DeviceContext for SYCL builds.
file	device_stream.cpp
	Implements the DeviceStream for CPU builds.
file	device_stream.h
	Declarations for DeviceStream class.
file	device_stream_hip.cpp
	Implements the DeviceStream for HIP.
file	device_stream_manager.cpp
	Implements GPU stream manager.
file	device_stream_manager.h
	This file declares a manager of GPU context and streams needed for running workloads on GPUs.
file	device_stream_ocl.cpp
	Implements the DeviceStream for OpenCL.
file	device_stream_sycl.cpp
	Implements the DeviceStream for SYCL builds.
file	devicebuffer_sycl.cpp
	Implements the DeviceBuffer type and routines for SYCL.
file	gpu_kernel_utils.h
	NBNXM GPU kernel utility methods.
file	gputraits.h
	Declares the GPU type traits for non-GPU builds.
file	gputraits_hip.h
	Declares the HIP type traits.
file	gputraits_ocl.h
	Declares the OpenCL type traits.
file	gputraits_sycl.h
	Declares the SYCL type traits.
file	nvshmem_manager.cpp
	Definitions for NVSHMEM initialization/finalize class.
file	nvshmem_manager.h
	Declarations for NVSHMEM initialization/finalize class.
file	nvshmem_utils.cpp
	Definitions for NVSHMEM halo exchange helper class.
file	nvshmem_utils.h
	Declarations for NVSHMEM GPU halo exchange helper class.
file	device_buffer.cpp
	Tests for device buffer.
file	device_stream_manager.cpp
	Tests GPU stream manager.
file	gpueventsynchronizer.cpp
	Tests for GpuEventSynchronizer.
file	nbnxm_kernel_utils.h
	NBNXM GPU kernel utility methods.