ge_options

Applicable Products

Product	Supported (√/x)
Ascend 950PR / Ascend 950DT	√
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	x
Atlas training products	x

Note: For Atlas A2 training products / Atlas A2 inference products , only the Atlas 800I A2 inference server and A200I A2 Box heterogeneous subrack are supported.

Function Description

Configures additional GE configuration items.

Prototype

      
           ge_options(ge_options)

Parameters

Parameter	Data Type	Value Description
ge_options	dict[str, str]	Configures GE configuration items. Among these, ge.flowGraphMemMaxSize is a key parameter that specifies the maximum memory that can be occupied by all KV caches. Setting this value too high may reduce the memory available for the model, so it must be set based on actual requirements. For details, see "Options".

Parameter

Data Type

Value Description

ge_options

dict[str, str]

Configures GE configuration items.

Among these, ge.flowGraphMemMaxSize is a key parameter that specifies the maximum memory that can be occupied by all KV caches. Setting this value too high may reduce the memory available for the model, so it must be set based on actual requirements.

For details, see "Options".

Example

      
           from llm_datadist import LLMConfig
ge_options = {
    "ge.flowGraphMemMaxSize": "4106127360"
}
llm_config = LLMConfig()
llm_config.ge_options = ge_options

Returns

In normal cases, no value is returned.

If a parameter is incorrect, a TypeError or ValueError may be thrown.

Constraints

None

Parent topic: LLMConfig