REGIST_MATMUL_OBJ时如果没有传入tiling参数,可以通过该接口单独传入tiling参数,对matmul对象进行初始化。tiling参数的具体介绍请参考Matmul Tiling API。
__aicore__ inline void Init(const TCubeTiling* __restrict cubeTiling);
参数名 |
输入/输出 |
描述 |
---|---|---|
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);