SetFixPipeAddr

Product Support

Product

Supported

Atlas A3 training products/Atlas A3 inference products

x

Atlas A2 training products/Atlas A2 inference products

x

Atlas 200I/500 A2 inference products

Atlas inference product's AI Core

x

Atlas inference product's Vector Core

x

Atlas training products

x

Function

Sets the address of the LocalTensor for the element-wise operation after in-line quantization during DataCopy (CO1 -> GM).

Prototype

1
2
template <typename T>
__aicore__ inline void SetFixPipeAddr(const LocalTensor<T>& eleWiseData, uint16_t c0ChStride)

Parameters

Table 1 Parameters

Parameter

Input/Output

Meaning

eleWiseData

Input

Number of source operands on L1 Buffer. Type: LocalTensor.

Supported TPosition: A1/B1/C1. The start address must be 32-byte aligned and only the half data type is supported.

c0ChStride

Input

Stride of the C0 channel on the L1 Buffer. The unit is C0_SIZE (32 bytes).

The data chunk obtained by splitting eleWiseData in the unit of C0 along the N direction is called the C0 channel. The interval between two C0 channels is called the C0 channel stride.

Restrictions

None

Returns

None

Example

For details, see the complete example.

1
2
3
4
__aicore__inline void SetEleSrcPara(const LocalTensor <half>& eleWiseData, uint16_t c0ChStride)
{
    AscendC::SetFixPipeAddr(eleWiseData, c0ChStride);
}