#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.