KernelGenWgslIndex.cpp 文件参考

#include <algorithm>
#include <cmath>
#include <sstream>
#include "common/Exception.h"
#include "tensor/KernelGenWgslInternal.h"
#include "tensor/Quant.h"

浏览源代码.

命名空间
namespace	eve
	Build metadata (engine git commit, build time, third-party version).
namespace	eve::tensor
namespace	eve::tensor::wgsl_detail

函数
void	eve::tensor::wgsl_detail::genEmbedding (const Graph &, const FusedGroup &, KernelSpec &)
	Gen embedding.
void	eve::tensor::wgsl_detail::genConcat (const Graph &, const FusedGroup &, KernelSpec &)
	Gen concat.
void	eve::tensor::wgsl_detail::genSlice (const Graph &, const FusedGroup &, KernelSpec &)
	Gen slice.
void	eve::tensor::wgsl_detail::genPermute (const Graph &, const FusedGroup &, KernelSpec &)
	Gen permute.
void	eve::tensor::wgsl_detail::genResize2d (const Graph &, const FusedGroup &, KernelSpec &)
	Gen resize 2 d.
void	eve::tensor::wgsl_detail::genSdpa (const Graph &, const FusedGroup &, KernelSpec &)
	Gen sdpa.