device_id

Applicability

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

Note: For the 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. You can configure one or more device IDs. If you configure multiple device IDs, separate them with commas (,). In the single-process single-device scenario, set this parameter to a single value, for example, 0. In the single-process multi-device scenario, set this parameter to a list, for example, [0, 1].

Parameter

Data Type

Description

device_id

Union[int, List[int], Tuple[int]]

Device ID(s) of the current process. You can configure one or more device IDs. If you configure multiple device IDs, separate them with commas (,).

In the single-process single-device scenario, set this parameter to a single value, for example, 0.
In the single-process multi-device scenario, set this parameter to a list, for example, [0, 1].

Example

from llm_datadist import LLMConfig
llm_config = LLMConfig()
# Single-process single-device configuration method
llm_config.device_id = 0
# Single-process multi-device configuration method
# llm_config.device_id = [0, 1]

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