gpu_common.h File Reference

#include <cuda.h>
#include <cuda_runtime.h>
#include <cuda_runtime_api.h>
#include <iostream>
#include <unsupported/Eigen/CXX11/src/Tensor/TensorGpuHipCudaDefines.h>

Include dependency graph for gpu_common.h:

Go to the source code of this file.

Classes
struct	compile_time_device_info

Macros
#define	EIGEN_USE_GPU

Functions
void	ei_test_init_gpu ()
template<typename Kernel , typename Input , typename Output >
void	run_and_compare_to_gpu (const Kernel &ker, int n, const Input &in, Output &out)
template<typename Kernel , typename Input , typename Output >
void	run_on_cpu (const Kernel &ker, int n, const Input &in, Output &out)
template<typename Kernel , typename Input , typename Output >
void	run_on_gpu (const Kernel &ker, int n, const Input &in, Output &out)
template<typename Kernel , typename Input , typename Output >
__global__ EIGEN_HIP_LAUNCH_BOUNDS_1024 void	run_on_gpu_meta_kernel (const Kernel ker, int n, const Input *in, Output *out)

Variables
dim3	blockDim
dim3	blockIdx
dim3	threadIdx

Macro Definition Documentation

EIGEN_USE_GPU

#define EIGEN_USE_GPU

Definition at line 15 of file gpu_common.h.

Function Documentation

ei_test_init_gpu()

void ei_test_init_gpu

(

)

Definition at line 131 of file gpu_common.h.

run_and_compare_to_gpu()

template<typename Kernel , typename Input , typename Output >

void run_and_compare_to_gpu	(	const Kernel &	ker,
		int	n,
		const Input &	in,
		Output &	out
	)

Definition at line 96 of file gpu_common.h.

run_on_cpu()

template<typename Kernel , typename Input , typename Output >

void run_on_cpu	(	const Kernel &	ker,
		int	n,
		const Input &	in,
		Output &	out
	)

Definition at line 23 of file gpu_common.h.

run_on_gpu()

template<typename Kernel , typename Input , typename Output >

void run_on_gpu	(	const Kernel &	ker,
		int	n,
		const Input &	in,
		Output &	out
	)

Definition at line 43 of file gpu_common.h.

run_on_gpu_meta_kernel()

template<typename Kernel , typename Input , typename Output >

__global__ EIGEN_HIP_LAUNCH_BOUNDS_1024 void run_on_gpu_meta_kernel	(	const Kernel	ker,
		int	n,
		const Input *	in,
		Output *	out
	)

Definition at line 33 of file gpu_common.h.

Variable Documentation

blockDim

dim3 blockDim

Definition at line 19 of file gpu_common.h.

blockIdx

dim3 blockIdx

Definition at line 19 of file gpu_common.h.

threadIdx

dim3 threadIdx

Definition at line 19 of file gpu_common.h.