gemm_softmax.cpp File Reference

#include "tests/utils/utils.hpp"
#include "xetla.hpp"

Include dependency graph for gemm_softmax.cpp:

Macros
#define	SIMD   32

Functions
template<typename data_type_a , typename data_type_b , typename data_type_c , typename data_type_acc = float>
int	gemm_softmax_result_validate (data_type_a *A_device, data_type_b *B_device, data_type_c *C_device, uint32_t m, uint32_t k, uint32_t n, uint32_t batch_num, sycl::queue &queue, mem_layout mem_layout_a_=mem_layout::row_major, mem_layout mem_layout_b_=mem_layout::row_major)
void	gemm_softmax_run (uint32_t iter)
int	main ()

Macro Definition Documentation

SIMD

#define SIMD   32

Function Documentation

gemm_softmax_result_validate()

template<typename data_type_a , typename data_type_b , typename data_type_c , typename data_type_acc = float>

int gemm_softmax_result_validate	(	data_type_a *	A_device,
		data_type_b *	B_device,
		data_type_c *	C_device,
		uint32_t	m,
		uint32_t	k,
		uint32_t	n,
		uint32_t	batch_num,
		sycl::queue &	queue,
		mem_layout	mem_layout_a_ = mem_layout::row_major,
		mem_layout	mem_layout_b_ = mem_layout::row_major
	)

gemm_softmax_run()

void gemm_softmax_run

(

uint32_t

iter

)

main()

int main

(

)