device_id

Applicability

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

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

Function Description

Sets the device ID of the current process, corresponding to the underlying ge.exec.deviceId configuration item.

Prototype

device_id(device_id)

Parameters

Parameter	Data Type	Description
device_id	Union[int, List[int], Tuple[int]]	Device ID(s) of the current process. Currently, only one device ID can be configured.

Example

from llm_datadist import LLMConfig
llm_config = LLMConfig()
llm_config.device_id = 0

Returns

In normal cases, no value is returned.

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

Restrictions

None

Parent topic: LLMConfig