SetFixPipeAddr

Supported Products

Product

Supported (√/x)

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 Usage

Sets the address of the LocalTensor during the elementwise operation after on-the-fly quantization is performed during the DataCopy (CO1->GM) process.

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 eleWiseTensor 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

Return Value Description

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);
}