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.

Functions
template<typename T >
__global__ void	pushShmemKernel (T *dest, const T *__restrict__ src, const size_t n, const int destRank, uint64_t iter, uint64_t *doneSig, uint64_t *readySlot)
template<>
__global__ void	pushShmemKernel (float *dest, const float *__restrict__ src, const size_t n, const int destRank, uint64_t iter, uint64_t *doneSig, uint64_t *readySlot)
template<>
__global__ void	pushShmemKernel (double *dest, const double *__restrict__ src, const size_t n, const int destRank, uint64_t iter, uint64_t *doneSig, uint64_t *readySlot)
__global__ void	pushShmemKernelWait (uint64_t iter, uint64_t *doneSlot)
__global__ void	postReadyShmemKernel (uint64_t *readySlot, uint64_t iter, const int srcRank)

Function Documentation

postReadyShmemKernel()

__global__ void postReadyShmemKernel	(	uint64_t *	readySlot,
		uint64_t	iter,
		const int	srcRank
	)

Definition at line 139 of file gs_nvshmem_kernels.h.

Here is the call graph for this function:

pushShmemKernel() [1/3]

template<>

__global__ void pushShmemKernel	(	double *	dest,
		const double *__restrict__	src,
		const size_t	n,
		const int	destRank,
		uint64_t	iter,
		uint64_t *	doneSig,
		uint64_t *	readySlot
	)

Definition at line 105 of file gs_nvshmem_kernels.h.

Here is the call graph for this function:

pushShmemKernel() [2/3]

template<>

__global__ void pushShmemKernel	(	float *	dest,
		const float *__restrict__	src,
		const size_t	n,
		const int	destRank,
		uint64_t	iter,
		uint64_t *	doneSig,
		uint64_t *	readySlot
	)

Definition at line 83 of file gs_nvshmem_kernels.h.

Here is the call graph for this function:

pushShmemKernel() [3/3]

template<typename T >

__global__ void pushShmemKernel	(	T *	dest,
		const T *__restrict__	src,
		const size_t	n,
		const int	destRank,
		uint64_t	iter,
		uint64_t *	doneSig,
		uint64_t *	readySlot
	)

pushShmemKernelWait()

__global__ void pushShmemKernelWait	(	uint64_t	iter,
		uint64_t *	doneSlot
	)

Definition at line 128 of file gs_nvshmem_kernels.h.

Here is the call graph for this function: