device_id

Applicability

Product	Supported
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 the Atlas A2 training products/Atlas A2 inference products, only the Atlas 800I A2 inference server, Atlas 300I A2 inference card, and A200I A2 Box heterogeneous components 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