CheckNodeSupportOnAicore

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	√

Header File/Library File

Header file: #include <ge/ge_utils.h>
Library file: libge_compiler.so

Function Usage

Verifies the input node to check whether it can be executed on the AI Core.

Prototype

static Status CheckNodeSupportOnAicore(const GNode &node, bool &is_supported, AscendString &unsupported_reason)

Parameters

Parameter	Input/Output	Description
node	Input	Node to be verified.
is_supported	Output	Whether the node can be executed on the AI Core. true: Supported. false: Not supported.
unsupported_reason	Output	Reason why the node execution is not supported.

Parameter

Input/Output

Description

node

Input

Node to be verified.

is_supported

Output

Whether the node can be executed on the AI Core.

true: Supported.
false: Not supported.

unsupported_reason

Output

Reason why the node execution is not supported.

Returns

Parameter	Type	Description
-	graphStatus	SUCCESS: Verification succeeded. FAILED: Verification failed.

Parameter

Type

Description

graphStatus

SUCCESS: Verification succeeded.

FAILED: Verification failed.

Restrictions

Before calling this API, ensure that the shape of the node has been inferred. This is because the verification process requires the shape information.

Parent topic: GeUtils