#include <dudxyz_kernel.h>

Static Public Member Functions
static __device__ void	run (T , const T , const T , const T , const T , const T , const T , const T , const T *, const int)

Detailed Description

template<typename T, const int LX, const int NWF>
struct dudxyz_mfma_dispatch< T, LX, NWF >

Device kernel for the derivative on the AMD matrix cores

The first half of ax_helm_mfma_elem() and nothing else: the same four LDS cubes and the same three D * U contractions handed to mfma_contract_sel, but the geometric factors turn the reference derivatives into the physical one pointwise on the way out to global memory, where axhelm contracts a second time with D^T. See mfma_kernel.h for the tiles, the lane layouts, the wavefronts-per-element scheme and the arch and LX bounds.

Definition at line 322 of file dudxyz_kernel.h.

Member Function Documentation

◆ run()

template<typename T , const int LX, const int NWF>

static __device__ void dudxyz_mfma_dispatch< T, LX, NWF >::run	(	T *	,
		const T *	,
		const T *	,
		const T *	,
		const T *	,
		const T *	,
		const T *	,
		const T *	,
		const T *	,
		const int
	)

inlinestatic

Definition at line 323 of file dudxyz_kernel.h.

Here is the caller graph for this function:

The documentation for this struct was generated from the following file:

/home/runner/work/neko/neko/src/math/bcknd/device/hip/dudxyz_kernel.h

Static Public Member Functions

Detailed Description

Member Function Documentation

◆ run()