文档
注册
评分
提单
论坛
小AI

REGIST_MATMUL_OBJ

功能说明

初始化Matmul对象。

函数原型

1
REGIST_MATMUL_OBJ(tpipe, workspace, ...)

参数说明

表1 参数说明

参数名

输入/输出

描述

tpipe

输入

Tpipe对象。

workspace

输入

系统workspace指针。

...

输入

可变参数,传入Matmul对象和与之对应的Tiling结构,要求Tiling结构的数据类型为TCubeTiling结构

Tiling参数可以通过host侧GetTiling接口获取,并传递到kernel侧使用。

返回值

无。

注意事项

  • 分离架构中,本接口必须在InitBuffer接口前调用。
  • 在程序中,最多支持定义4个Matmul对象。

调用示例

// 推荐:初始化单个matmul对象,传入tiling参数
REGIST_MATMUL_OBJ(&pipe, GetSysWorkSpacePtr(), mm, &tiling);
// 推荐:初始化多个matmul对象,传入对应的tiling参数
REGIST_MATMUL_OBJ(&pipeIn, GetSysWorkSpacePtr(), mm1, mm1tiling, mm2, mm2tiling, mm3, mm3tiling, mm4, mm4tiling);
// 初始化单个matmul对象,未传入tiling参数。注意,该场景下需要使用Init接口单独传入tiling参数。这种方式将matmul对象的初始化和tiling的设置分离,比如,Tiling可变的场景,可通过这种方式多次对Tiling进行重新设置
REGIST_MATMUL_OBJ(&pipe, GetSysWorkSpacePtr(), mm);
mm.Init(&tiling);
搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词