Init

功能说明

REGIST_MATMUL_OBJ时如果没有传入tiling参数,可以通过该接口单独传入tiling参数,对matmul对象进行初始化。tiling参数的具体介绍请参考Matmul Tiling API

函数原型

__aicore__ inline void Init(const TCubeTiling* __restrict cubeTiling);

参数说明

表1 接口参数说明

参数名

输入/输出

描述

cubeTiling

输入

Matmul tiling参数,TCubeTiling结构体定义请参见表2 TCubeTiling结构说明

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

返回值

支持的型号

Atlas A2训练系列产品

Atlas推理系列产品(Ascend 310P处理器)AI Core

注意事项

调用示例

// 不建议:初始化单个matmul对象,未传入tiling参数。注意,该场景下需要使用Init接口单独传入tiling参数,对matmul对象进行初始化。
REGIST_MATMUL_OBJ(&pipe, GetSysWorkSpacePtr(), mm);
mm.Init(&tiling);