UpdateGraphFeatureMemoryBase

Applicability

Product

Supported or Not

Atlas A3 training products/Atlas A3 inference products

Atlas A2 training products/Atlas A2 inference products

Atlas 200I/500 A2 inference products

x

Atlas inference products

Atlas training products

Header File/Library File

  • Header file: #include <ge/ge_api.h>
  • Library file: libge_runner.so

Function Usage

Updates the feature memory base address of a graph.

The memory size is obtained from GetCompiledGraphSummary > GetFeatureMemorySize.

Feature memory refers to the intermediate memory required during model execution (for example, the input and output memories of intermediate nodes).

Prototype

1
Status UpdateGraphFeatureMemoryBase(uint32_t graph_id, const void *const memory, size_t size)

Parameters

Parameter

Input/Output

Description

graph_id

Input

Subgraph ID.

memory

Input

Base address of the feature memory.

size

Input

Size of the feature memory.

Returns

Parameter

Type

Description

-

Status

SUCCESS: The setting is successful.

FAILED: The setting fails.

Restrictions