eve::tensor::wgsl_detail::ChainContext结构体 参考

Public 成员函数
std::string	operand (int nodeId)
std::string	emit (int nodeId)

Public 属性
const Graph &	graph
const FusedGroup &	group
const std::vector< std::string > &	indexExprs
std::string	rootVar
std::ostringstream &	os
int	temp = 0
std::string	biasIndexExpr
std::vector< std::string >	vars

详细描述

Emit the elementwise chain expression. rootVar is the value entering the first chain node (e.g. the matmul accumulator). Input fetches reference buffers by inputBind + "[" + indexExpr + "]". Returns the final variable.

在文件 KernelGenWgsl.cpp 第 105 行定义.

成员函数说明

emit()

std::string eve::tensor::wgsl_detail::ChainContext::emit ( int  nodeId )

inline

在文件 KernelGenWgsl.cpp 第 126 行定义.

引用了 eve::tensor::Abs, eve::tensor::Add, eve::tensor::AddScalar, eve::tensor::Clamp, eve::tensor::Cos, eve::tensor::Divide, eve::tensor::DivScalar, eve::tensor::Exp, eve::tensor::Gelu, graph, group, eve::tensor::GraphNode::in0, eve::tensor::GraphNode::in1, eve::tensor::GraphNode::in2, eve::tensor::Log, eve::tensor::MaximumScalar, eve::tensor::MinimumScalar, eve::tensor::MulScalar, eve::tensor::Multiply, eve::tensor::Neg, eve::tensor::Graph::node(), nodeId, eve::tensor::FusedGroup::nodes, operand(), os, eve::tensor::PowScalar, eve::tensor::Relu, rootVar, eve::tensor::GraphNode::s0, eve::tensor::GraphNode::s1, eve::tensor::wgsl_detail::scalarStr(), eve::tensor::Sigmoid, eve::tensor::Silu, eve::tensor::Sin, eve::tensor::Sqrt, eve::tensor::Sub, eve::tensor::SubScalar, eve::tensor::Tanh, temp, eve::tensor::GraphNode::type, vars, eve::tensor::Where , 以及 x.

operand()

std::string eve::tensor::wgsl_detail::ChainContext::operand ( int  nodeId )

inline

在文件 KernelGenWgsl.cpp 第 114 行定义.

引用了 biasIndexExpr, eve::tensor::FusedGroup::biasNode, group, idx, indexExprs, eve::tensor::FusedGroup::inputs, nodeId , 以及 vars.

被这些函数引用 emit().

类成员变量说明

biasIndexExpr

std::string eve::tensor::wgsl_detail::ChainContext::biasIndexExpr

在文件 KernelGenWgsl.cpp 第 112 行定义.

被这些函数引用 operand().

graph

const Graph& eve::tensor::wgsl_detail::ChainContext::graph

在文件 KernelGenWgsl.cpp 第 106 行定义.

被这些函数引用 emit().

group

const FusedGroup& eve::tensor::wgsl_detail::ChainContext::group

在文件 KernelGenWgsl.cpp 第 107 行定义.

被这些函数引用 emit() , 以及 operand().

indexExprs

const std::vector<std::string>& eve::tensor::wgsl_detail::ChainContext::indexExprs

在文件 KernelGenWgsl.cpp 第 108 行定义.

被这些函数引用 operand().

os

std::ostringstream& eve::tensor::wgsl_detail::ChainContext::os

在文件 KernelGenWgsl.cpp 第 110 行定义.

被这些函数引用 emit().

rootVar

std::string eve::tensor::wgsl_detail::ChainContext::rootVar

在文件 KernelGenWgsl.cpp 第 109 行定义.

被这些函数引用 emit().

temp

int eve::tensor::wgsl_detail::ChainContext::temp = 0

在文件 KernelGenWgsl.cpp 第 111 行定义.

被这些函数引用 emit().

vars

std::vector<std::string> eve::tensor::wgsl_detail::ChainContext::vars

在文件 KernelGenWgsl.cpp 第 124 行定义.

被这些函数引用 emit() , 以及 operand().

---

该结构体的文档由以下文件生成:

• src/modules/tensor/KernelGenWgsl.cpp