SetTail

Applicability

Product	Supported
Atlas A3 training products/Atlas A3 inference products	√
Atlas A2 training products/Atlas A2 inference products	√
Atlas 200I/500 A2 inference products	√
Atlas inference product's AI Core	√
Atlas inference product's Vector Core	x
Atlas training products	x

Function

Resets singleCoreM, singleCoreN, and singleCoreK for the current computation without changing tiling. The unit is element.

Prototype

__aicore__ inline void SetTail(int tailM = -1, int tailN = -1, int tailK = -1)

Parameters

Parameter	Input/Output	Description
tailM	Input	singleCoreM value after resetting
tailN	Input	singleCoreN value after resetting
tailK	Input	singleCoreK value after resetting

Returns

None

Restrictions

None

Example

REGIST_MATMUL_OBJ(&pipe, GetSysWorkSpacePtr(), mm, &tiling);
mm.SetTensorA(gm_a);
mm.SetTensorB(gm_b);
mm.SetBias(gm_bias);
mm.SetTail(tailM,tailN,tailK);    // If it is a tail core, adjustments need to be made to singleCoreM, singleCoreN, and singleCoreK.
mm.IterateAll(gm_c);

Parent topic: Matmul Kernel APIs