KernelGenWgslInternal.h 文件参考

#include "tensor/KernelGenWgsl.h"

浏览源代码.

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

函数
std::string	eve::tensor::wgsl_detail::header (int localX, int localY=1)
	Header.
std::string	eve::tensor::wgsl_detail::bufferDecl (int binding, const char *name)
	Buffer decl.
std::string	eve::tensor::wgsl_detail::bufferDeclUint (int binding, const char *name)
	Buffer decl uint.
std::string	eve::tensor::wgsl_detail::pushConstant ()
	Pushes constant.
std::string	eve::tensor::wgsl_detail::scalarStr (float value)
	Scalar str.
std::string	eve::tensor::wgsl_detail::emitQuantizedBVal (DType dtype, int group, const char *bufferName)
	Emit quantized b val.
int	eve::tensor::wgsl_detail::groupsFor (int count)
	Groups for.
void	eve::tensor::wgsl_detail::specializeInputBindings (const Graph &, const FusedGroup &, KernelSpec &)
	Specialize input bindings.
void	eve::tensor::wgsl_detail::genSoftmax (const Graph &, const FusedGroup &, KernelSpec &)
	Gen softmax.
void	eve::tensor::wgsl_detail::genNorm (const Graph &, const FusedGroup &, bool rms, KernelSpec &)
	Gen norm.
void	eve::tensor::wgsl_detail::genReduceOrArgmax (const Graph &, const FusedGroup &, bool argmax, KernelSpec &)
	Gen reduce or argmax.
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.

变量
constexpr int	eve::tensor::wgsl_detail::kLocalSize = 256