pipecg_kernel.h File Reference

#include <math/bcknd/device/hip/math_kernel.h>

Include dependency graph for pipecg_kernel.h:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Functions
template<typename T >
__global__ void	cg_update_xp_kernel (T *__restrict__ x, T *__restrict__ p, T **__restrict__ u, const T *alpha, const T *beta, const int p_cur, const int p_space, const int n)
template<typename T >
__global__ void	pipecg_vecops_kernel (T *__restrict__ p, T *__restrict__ q, T *__restrict__ r, T *__restrict__ s, T *__restrict__ u1, T *__restrict__ u2, T *__restrict__ w, T *__restrict__ z, T *__restrict__ ni, T *__restrict__ mi, const T alpha, const T beta, const T *mult, T *buf_h1, T *buf_h2, T *buf_h3, const int n)

Function Documentation

cg_update_xp_kernel()

template<typename T >

__global__ void cg_update_xp_kernel	(	T *__restrict__	x,
		T *__restrict__	p,
		T **__restrict__	u,
		const T *	alpha,
		const T *	beta,
		const int	p_cur,
		const int	p_space,
		const int	n
	)

Kernel for back-substitution of x and update of p

Definition at line 43 of file pipecg_kernel.h.

pipecg_vecops_kernel()

template<typename T >

__global__ void pipecg_vecops_kernel	(	T *__restrict__	p,
		T *__restrict__	q,
		T *__restrict__	r,
		T *__restrict__	s,
		T *__restrict__	u1,
		T *__restrict__	u2,
		T *__restrict__	w,
		T *__restrict__	z,
		T *__restrict__	ni,
		T *__restrict__	mi,
		const T	alpha,
		const T	beta,
		const T *	mult,
		T *	buf_h1,
		T *	buf_h2,
		T *	buf_h3,
		const int	n
	)

Device kernel for pipecg_vecops

Definition at line 73 of file pipecg_kernel.h.

Here is the call graph for this function: