SetFixPipeConfig

Product Support

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	x
Atlas inference product's Vector Core	x
Atlas training products	x

Function

Sets the tensor quantization parameters for in-line quantization during DataCopy (CO1->GM or CO1->A1).

Prototype

template <typename T>
__aicore__ inline void SetFixPipeConfig(const LocalTensor<T>& reluPre, const LocalTensor<T>& quantPre, bool isUnitFlag = false)
template <typename T, bool setRelu = false>
__aicore__ inline void SetFixPipeConfig(const LocalTensor<T>& preData, bool isUnitFlag = false)

Parameters

**Table 1** Template parameters
Parameter	Description
T	Data type of the operand.
setRelu	For one tensor, when setRelu is set to true, reluPre needs to be set. Otherwise, quantPre needs to be set. Currently, only false is supported.

**Table 2** Parameters
Parameter	Input/Output	Meaning
reluPre	Input	Source operand, ReLU tensor. Type: LocalTensor. Supported TPosition: C2PIPE2GM. reluPre is reserved for future use. You can pass an empty LocalTensor.
quantPre	Input	Source operand, quantization tensor, a tensor involved in compute during quantization. Type: LocalTensor. Supported TPosition: C2PIPE2GM.
isUnitFlag	Input	UnitFlag configuration item. The type is bool. Reserved for future use. Retain the default value false.
preData	Input	Used to set a tensor. A switch can be used to control whether the tensor is a ReLU tensor or quantization tensor. The supported TPosition is C2PIPE2GM. Currently, only the quantization tensor can be passed.

Restrictions

quantPre and reluPre must be tensors on the Fixpipe Buffer.

Returns

None

Example

For details, see the complete example.

__aicore__inline void SetFPC(const LocalTensor <int32_t>& reluPreTensor, const LocalTensor <int32_t>& quantPreTensor)
{
    // reluPreTensor is an empty tensor.
    AscendC::SetFixPipeConfig<int32_t>(reluPreTensor, quantPreTensor);

    // Equivalent calling:
    // AscendC::SetFixPipeConfig<int32_t>(quantPreTensor);
}

Parent topic: Data Movement