Op2StageExecuteFuncs

Function Usage

Sets operator-level callback functions in scenarios where a single operator contains multi-kernel execution logic. The setting includes the delivery preparation callback function and delivery callback function. The delivery preparation callback function prepares parameters to be used for multi-kernel delivery, and the delivery callback function implements this delivery. This is an internal associated API and is not directly called by developers. Therefore, you do not need to pay attention to it.

Prototype

1

OpImplRegisterV2 &Op2StageExecuteFuncs(OpExecPrepareFunc prepare_func, OpExecLaunchFunc launch_func)

Parameters

1

using OpExecPrepareFunc = UINT32 (*)(OpExecutePrepareContext *);

1

using OpExecLaunchFunc = UINT32 (*)(OpExecuteLaunchContext *);

Returns

OpImplRegisterV2 object of the operator, where the operator-level callback function is registered.

Constraints

None