载入中...
搜索中...
未找到
Public 属性 | 所有成员列表
eve::tensor::KernelSpec结构体 参考

KernelSpec public API. 更多...

#include <KernelGen.h>

Public 属性

std::string pass1
 
std::string pass2
 
int groupsX1 = 0
 
int groupsY1 = 1
 
int groupsZ1 = 1
 
int groupsX2 = 0
 
int groupsY2 = 1
 
int groupsZ2 = 1
 
std::vector< int > inputRepresentatives
 Input slot to representative slot; empty means identity. WGSL aliases repeated read inputs to one declaration so WebGPU never binds overlapping writable ranges.
 
int inputCount = 0
 
int inputsReadPass1 = 0
 
int statsCount = 0
 
int statsSize = 0
 
bool twoPass = false
 
int qDtype = 0
 
int qGroup = 0
 
int scalesBinding = -1
 
int outputBinding = -1
 

详细描述

KernelSpec public API.

A generated, fully specialized compute kernel (AITemplate-style codegen).

Shapes are baked into the GLSL or WGSL as constants; the only runtime inputs are the storage buffers themselves, so a CompiledFunction is a static executable program with no per-dispatch shape computation.

Binding layout:

  • single-pass kernels: inputs 0..k-1, output binding k;
  • two-pass kernels (softmax / layernorm / rmsnorm): pass1 binds the first inputsReadPass1 inputs + statsCount stats buffers; pass2 binds all inputs + stats + the output as the last binding.

在文件 KernelGen.h 第 27 行定义.

类成员变量说明

◆ groupsX1

int eve::tensor::KernelSpec::groupsX1 = 0

在文件 KernelGen.h 第 30 行定义.

被这些函数引用 eve::tensor::wgsl_detail::genNorm(), eve::tensor::glsl_detail::genNorm(), eve::tensor::wgsl_detail::genSoftmax() , 以及 eve::tensor::glsl_detail::genSoftmax().

◆ groupsX2

int eve::tensor::KernelSpec::groupsX2 = 0

在文件 KernelGen.h 第 31 行定义.

被这些函数引用 eve::tensor::wgsl_detail::genConcat(), eve::tensor::glsl_detail::genConcat(), eve::tensor::glsl_detail::genConv(), eve::tensor::wgsl_detail::genConv(), eve::tensor::glsl_detail::genElementwise(), eve::tensor::wgsl_detail::genElementwise(), eve::tensor::wgsl_detail::genEmbedding(), eve::tensor::glsl_detail::genEmbedding(), eve::tensor::glsl_detail::genMatMul(), eve::tensor::wgsl_detail::genMatMul(), eve::tensor::wgsl_detail::genNorm(), eve::tensor::glsl_detail::genNorm(), eve::tensor::wgsl_detail::genPermute(), eve::tensor::glsl_detail::genPermute(), eve::tensor::glsl_detail::genPool(), eve::tensor::wgsl_detail::genPool(), eve::tensor::wgsl_detail::genReduceOrArgmax(), eve::tensor::glsl_detail::genReduceOrArgmax(), eve::tensor::glsl_detail::genResize2d(), eve::tensor::wgsl_detail::genResize2d(), eve::tensor::wgsl_detail::genSdpa(), eve::tensor::glsl_detail::genSdpa(), eve::tensor::wgsl_detail::genSlice(), eve::tensor::glsl_detail::genSlice(), eve::tensor::wgsl_detail::genSoftmax(), eve::tensor::glsl_detail::genSoftmax() , 以及 eve::tensor::GpuProgram::tryBuild().

◆ groupsY1

int eve::tensor::KernelSpec::groupsY1 = 1

在文件 KernelGen.h 第 30 行定义.

◆ groupsY2

int eve::tensor::KernelSpec::groupsY2 = 1

在文件 KernelGen.h 第 31 行定义.

被这些函数引用 eve::tensor::glsl_detail::genMatMul(), eve::tensor::wgsl_detail::genMatMul(), eve::tensor::wgsl_detail::genSdpa(), eve::tensor::glsl_detail::genSdpa() , 以及 eve::tensor::GpuProgram::tryBuild().

◆ groupsZ1

int eve::tensor::KernelSpec::groupsZ1 = 1

在文件 KernelGen.h 第 30 行定义.

◆ groupsZ2

int eve::tensor::KernelSpec::groupsZ2 = 1

在文件 KernelGen.h 第 31 行定义.

◆ inputCount

int eve::tensor::KernelSpec::inputCount = 0

在文件 KernelGen.h 第 35 行定义.

被这些函数引用 eve::tensor::wgsl_detail::genConcat(), eve::tensor::glsl_detail::genConcat(), eve::tensor::glsl_detail::genConv(), eve::tensor::wgsl_detail::genConv(), eve::tensor::glsl_detail::genElementwise(), eve::tensor::wgsl_detail::genElementwise(), eve::tensor::wgsl_detail::genEmbedding(), eve::tensor::glsl_detail::genEmbedding(), eve::tensor::glsl_detail::genMatMul(), eve::tensor::wgsl_detail::genMatMul(), eve::tensor::wgsl_detail::genNorm(), eve::tensor::glsl_detail::genNorm(), eve::tensor::wgsl_detail::genPermute(), eve::tensor::glsl_detail::genPermute(), eve::tensor::glsl_detail::genPool(), eve::tensor::wgsl_detail::genPool(), eve::tensor::wgsl_detail::genReduceOrArgmax(), eve::tensor::glsl_detail::genReduceOrArgmax(), eve::tensor::glsl_detail::genResize2d(), eve::tensor::wgsl_detail::genResize2d(), eve::tensor::wgsl_detail::genSdpa(), eve::tensor::glsl_detail::genSdpa(), eve::tensor::wgsl_detail::genSlice(), eve::tensor::glsl_detail::genSlice(), eve::tensor::wgsl_detail::genSoftmax(), eve::tensor::glsl_detail::genSoftmax() , 以及 eve::tensor::wgsl_detail::specializeInputBindings().

◆ inputRepresentatives

std::vector<int> eve::tensor::KernelSpec::inputRepresentatives

Input slot to representative slot; empty means identity. WGSL aliases repeated read inputs to one declaration so WebGPU never binds overlapping writable ranges.

在文件 KernelGen.h 第 34 行定义.

被这些函数引用 eve::tensor::wgsl_detail::specializeInputBindings().

◆ inputsReadPass1

int eve::tensor::KernelSpec::inputsReadPass1 = 0

在文件 KernelGen.h 第 36 行定义.

被这些函数引用 eve::tensor::wgsl_detail::genNorm(), eve::tensor::glsl_detail::genNorm(), eve::tensor::wgsl_detail::genSoftmax(), eve::tensor::glsl_detail::genSoftmax() , 以及 eve::tensor::wgsl_detail::specializeInputBindings().

◆ outputBinding

int eve::tensor::KernelSpec::outputBinding = -1

在文件 KernelGen.h 第 46 行定义.

被这些函数引用 eve::tensor::wgsl_detail::genEmbedding(), eve::tensor::glsl_detail::genEmbedding(), eve::tensor::glsl_detail::genMatMul() , 以及 eve::tensor::wgsl_detail::genMatMul().

◆ pass1

std::string eve::tensor::KernelSpec::pass1

在文件 KernelGen.h 第 28 行定义.

被这些函数引用 eve::tensor::wgsl_detail::genConcat(), eve::tensor::glsl_detail::genConcat(), eve::tensor::glsl_detail::genConv(), eve::tensor::wgsl_detail::genConv(), eve::tensor::glsl_detail::genElementwise(), eve::tensor::wgsl_detail::genElementwise(), eve::tensor::wgsl_detail::genEmbedding(), eve::tensor::glsl_detail::genEmbedding(), eve::tensor::glsl_detail::genMatMul(), eve::tensor::wgsl_detail::genMatMul(), eve::tensor::wgsl_detail::genNorm(), eve::tensor::glsl_detail::genNorm(), eve::tensor::wgsl_detail::genPermute(), eve::tensor::glsl_detail::genPermute(), eve::tensor::glsl_detail::genPool(), eve::tensor::wgsl_detail::genPool(), eve::tensor::wgsl_detail::genReduceOrArgmax(), eve::tensor::glsl_detail::genReduceOrArgmax(), eve::tensor::glsl_detail::genResize2d(), eve::tensor::wgsl_detail::genResize2d(), eve::tensor::wgsl_detail::genSdpa(), eve::tensor::glsl_detail::genSdpa(), eve::tensor::wgsl_detail::genSlice(), eve::tensor::glsl_detail::genSlice(), eve::tensor::wgsl_detail::genSoftmax(), eve::tensor::glsl_detail::genSoftmax(), eve::tensor::wgsl_detail::specializeInputBindings() , 以及 eve::tensor::GpuProgram::tryBuild().

◆ pass2

std::string eve::tensor::KernelSpec::pass2

在文件 KernelGen.h 第 29 行定义.

被这些函数引用 eve::tensor::wgsl_detail::genConcat(), eve::tensor::glsl_detail::genConcat(), eve::tensor::glsl_detail::genConv(), eve::tensor::wgsl_detail::genConv(), eve::tensor::glsl_detail::genElementwise(), eve::tensor::wgsl_detail::genElementwise(), eve::tensor::wgsl_detail::genEmbedding(), eve::tensor::glsl_detail::genEmbedding(), eve::tensor::glsl_detail::genMatMul(), eve::tensor::wgsl_detail::genMatMul(), eve::tensor::wgsl_detail::genNorm(), eve::tensor::glsl_detail::genNorm(), eve::tensor::wgsl_detail::genPermute(), eve::tensor::glsl_detail::genPermute(), eve::tensor::glsl_detail::genPool(), eve::tensor::wgsl_detail::genPool(), eve::tensor::wgsl_detail::genReduceOrArgmax(), eve::tensor::glsl_detail::genReduceOrArgmax(), eve::tensor::glsl_detail::genResize2d(), eve::tensor::wgsl_detail::genResize2d(), eve::tensor::wgsl_detail::genSdpa(), eve::tensor::glsl_detail::genSdpa(), eve::tensor::wgsl_detail::genSlice(), eve::tensor::glsl_detail::genSlice(), eve::tensor::wgsl_detail::genSoftmax(), eve::tensor::glsl_detail::genSoftmax(), eve::tensor::wgsl_detail::specializeInputBindings() , 以及 eve::tensor::GpuProgram::tryBuild().

◆ qDtype

int eve::tensor::KernelSpec::qDtype = 0

在文件 KernelGen.h 第 43 行定义.

被这些函数引用 eve::tensor::wgsl_detail::genEmbedding(), eve::tensor::glsl_detail::genEmbedding(), eve::tensor::glsl_detail::genMatMul() , 以及 eve::tensor::wgsl_detail::genMatMul().

◆ qGroup

int eve::tensor::KernelSpec::qGroup = 0

在文件 KernelGen.h 第 44 行定义.

被这些函数引用 eve::tensor::wgsl_detail::genEmbedding(), eve::tensor::glsl_detail::genEmbedding(), eve::tensor::glsl_detail::genMatMul() , 以及 eve::tensor::wgsl_detail::genMatMul().

◆ scalesBinding

int eve::tensor::KernelSpec::scalesBinding = -1

在文件 KernelGen.h 第 45 行定义.

被这些函数引用 eve::tensor::wgsl_detail::genEmbedding(), eve::tensor::glsl_detail::genEmbedding(), eve::tensor::glsl_detail::genMatMul() , 以及 eve::tensor::wgsl_detail::genMatMul().

◆ statsCount

int eve::tensor::KernelSpec::statsCount = 0

在文件 KernelGen.h 第 37 行定义.

被这些函数引用 eve::tensor::wgsl_detail::genNorm(), eve::tensor::glsl_detail::genNorm(), eve::tensor::wgsl_detail::genSoftmax(), eve::tensor::glsl_detail::genSoftmax() , 以及 eve::tensor::GpuProgram::tryBuild().

◆ statsSize

int eve::tensor::KernelSpec::statsSize = 0

在文件 KernelGen.h 第 38 行定义.

被这些函数引用 eve::tensor::wgsl_detail::genNorm(), eve::tensor::glsl_detail::genNorm(), eve::tensor::wgsl_detail::genSoftmax(), eve::tensor::glsl_detail::genSoftmax() , 以及 eve::tensor::GpuProgram::tryBuild().

◆ twoPass

bool eve::tensor::KernelSpec::twoPass = false

在文件 KernelGen.h 第 39 行定义.

被这些函数引用 eve::tensor::wgsl_detail::genNorm(), eve::tensor::glsl_detail::genNorm(), eve::tensor::wgsl_detail::genSoftmax(), eve::tensor::glsl_detail::genSoftmax(), eve::tensor::wgsl_detail::specializeInputBindings() , 以及 eve::tensor::GpuProgram::tryBuild().

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