gs_nvshmem_kernels.h File Reference

#include <nvshmemx.h>

Include dependency graph for gs_nvshmem_kernels.h:

Go to the source code of this file.

Macros
#define	__GS_NVHSMEM_KERNELS__

Functions
template<typename T >
__global__ void	pack_pushShmemKernel (const T *__restrict__ u, T *dest, T *__restrict__ src, const int *__restrict__ dof, const int destRank, const int srcRank, const int n, uint64_t counter, uint64_t *notifyDone, uint64_t *notifyReady)
template<>
__global__ void	pack_pushShmemKernel (const float *__restrict__ u, float *dest, float *__restrict__ src, const int *__restrict__ dof, const int destRank, const int srcRank, const int n, uint64_t counter, uint64_t *notifyDone, uint64_t *notifyReady)
template<>
__global__ void	pack_pushShmemKernel (const double *__restrict__ u, double *dest, double *__restrict__ src, const int *__restrict__ dof, const int destRank, const int srcRank, const int n, uint64_t counter, uint64_t *notifyDone, uint64_t *notifyReady)
__global__ void	pushShmemKernelWait (uint64_t counter, uint64_t *notifyDone)

Macro Definition Documentation

__GS_NVHSMEM_KERNELS__

#define __GS_NVHSMEM_KERNELS__

Definition at line 36 of file gs_nvshmem_kernels.h.

Function Documentation

pack_pushShmemKernel() [1/3]

template<>

__global__ void pack_pushShmemKernel	(	const double *__restrict__	u,
		double *	dest,
		double *__restrict__	src,
		const int *__restrict__	dof,
		const int	destRank,
		const int	srcRank,
		const int	n,
		uint64_t	counter,
		uint64_t *	notifyDone,
		uint64_t *	notifyReady
	)

Definition at line 99 of file gs_nvshmem_kernels.h.

Here is the call graph for this function:

pack_pushShmemKernel() [2/3]

template<>

__global__ void pack_pushShmemKernel	(	const float *__restrict__	u,
		float *	dest,
		float *__restrict__	src,
		const int *__restrict__	dof,
		const int	destRank,
		const int	srcRank,
		const int	n,
		uint64_t	counter,
		uint64_t *	notifyDone,
		uint64_t *	notifyReady
	)

Definition at line 54 of file gs_nvshmem_kernels.h.

Here is the call graph for this function:

pack_pushShmemKernel() [3/3]

template<typename T >

__global__ void pack_pushShmemKernel	(	const T *__restrict__	u,
		T *	dest,
		T *__restrict__	src,
		const int *__restrict__	dof,
		const int	destRank,
		const int	srcRank,
		const int	n,
		uint64_t	counter,
		uint64_t *	notifyDone,
		uint64_t *	notifyReady
	)

pushShmemKernelWait()

__global__ void pushShmemKernelWait	(	uint64_t	counter,
		uint64_t *	notifyDone
	)

Definition at line 143 of file gs_nvshmem_kernels.h.

Here is the call graph for this function: