SetTail

Function Description

Resets singleCoreM, singleCoreN, and singleCoreK for the current computation without changing tiling.

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

Availability

Precautions

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