aclopRegisterCompileFunc

Applicability

Product	Supported
Atlas 350 Accelerator Card	√
Atlas A3 training product/Atlas A3 inference product	√
Atlas A2 training product/Atlas A2 inference product	√
Atlas 200I/500 A2 inference product	√
Atlas inference product	√
Atlas training product	√

Function Usage

Registers an operator selector for a dynamic-shape operator. The selection of the tiling policy depends on the operator's input shape.

An operator can register with only one operator selector. To change the operator selector, call aclopUnregisterCompileFunc to unregister and then call aclopRegisterCompileFunc to register again.

Prototype

aclError aclopRegisterCompileFunc(const char *opType, aclopCompileFunc func)

Parameters

Parameter	Input/Output	Description
opType	Input	Pointer to the operator type.
func	Input	Callback function of the operator selector. typedef aclError (aclopCompileFunc)(int numInputs, const aclTensorDesc const inputDesc[], int numOutputs, const aclTensorDesc const outputDesc[], const aclopAttr opAttr, aclopKernelDesc *aclopKernelDesc);

Parameter

Input/Output

Description

opType

Input

Pointer to the operator type.

func

Input

Callback function of the operator selector.

typedef aclError (*aclopCompileFunc)(int numInputs, const aclTensorDesc *const inputDesc[], int numOutputs, const aclTensorDesc *const outputDesc[], const aclopAttr *opAttr, aclopKernelDesc *aclopKernelDesc);

Returns

0 on success; else, failure. For details, see aclError.

Parent topic: Single-Operator Model Execution