device_id

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

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	Value 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.

Constraints

None

Parent topic: LLMConfig