Neko 1.99.1
A portable framework for high-order spectral element flow simulations
Loading...
Searching...
No Matches
amg_cheby.hip File Reference
#include <hip/hip_runtime.h>
#include <device/device_config.h>
#include <device/hip/check.h>

Go to the source code of this file.

Functions

template<typename T >
__global__ void amg_cheby_solve_part1 (T *__restrict__ r, T *__restrict__ f, T *__restrict__ w, T *__restrict__ x, T *__restrict__ d, const T inv_thet, const bool zero_initial, const int n)
 
template<typename T >
__global__ void amg_cheby_solve_part2 (T *__restrict__ r, T *__restrict__ w, T *__restrict__ d, T *__restrict__ x, const T tmp1, const T tmp2, const int n)
 
void hip_amg_cheby_solve_part1 (void *r, void *f, void *w, void *x, void *d, real *inv_thet, int *n, bool *zero_initial, hipStream_t strm)
 
void hip_amg_cheby_solve_part2 (void *r, void *w, void *d, void *x, real *tmp1, real *tmp2, int *n, hipStream_t strm)
 

Function Documentation

◆ amg_cheby_solve_part1()

template<typename T >
__global__ void amg_cheby_solve_part1 ( T *__restrict__  r,
T *__restrict__  f,
T *__restrict__  w,
T *__restrict__  x,
T *__restrict__  d,
const T  inv_thet,
const bool  zero_initial,
const int  n 
)

Definition at line 40 of file amg_cheby.hip.

Here is the call graph for this function:

◆ amg_cheby_solve_part2()

template<typename T >
__global__ void amg_cheby_solve_part2 ( T *__restrict__  r,
T *__restrict__  w,
T *__restrict__  d,
T *__restrict__  x,
const T  tmp1,
const T  tmp2,
const int  n 
)

Definition at line 62 of file amg_cheby.hip.

Here is the call graph for this function:

◆ hip_amg_cheby_solve_part1()

void hip_amg_cheby_solve_part1 ( void r,
void f,
void w,
void x,
void d,
real inv_thet,
int n,
bool zero_initial,
hipStream_t  strm 
)

Fortran wrapper for part1 of cheby solve

Definition at line 86 of file amg_cheby.hip.

Here is the call graph for this function:

◆ hip_amg_cheby_solve_part2()

void hip_amg_cheby_solve_part2 ( void r,
void w,
void d,
void x,
real tmp1,
real tmp2,
int n,
hipStream_t  strm 
)

Fortran wrapper for part2 of cheby solve

Definition at line 102 of file amg_cheby.hip.

Here is the call graph for this function: