SetStartPosition

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

x

Atlas inference product's AI Core

x

Atlas inference product's Vector Core

x

Atlas training products

x

Function

Sets the start position for loading GradOutput data on a single core.

Prototype

1
__aicore__ inline void SetStartPosition(uint32_t hoStartIdx)

Parameters

Parameter

Input/Output

Description

hoStartIdx

Input

Start position in the GradOutput height direction of the current core.

Returns

None

Restrictions

None

Example

1
2
3
4
...
gradWeight_.SetSingleShape(singleShapeM, singleShapeN, singleShapeK);
gradWeight_.SetStartPosition(hoStartIdx); // Set the start position for loading GradOutput on a single core.
...