ModelConfiguration

class onnx_ir.ModelConfiguration(name, num_devices, device_names=())

A model-level set of devices that a multi-device plan runs on.

A model declares one or more configurations (mirroring ONNX’s DeviceConfigurationProto). Each one names a group of devices; the per-node NodeDeviceConfiguration entries then reference a configuration by object identity to say how that node is sharded or staged across those devices.

Device indices used elsewhere (for example ShardingSpec.device and onnx_ir.Node.shard()) are 0-based positions into device_names.

Added in version 1.0.0.

Parameters:

• name (str)

• num_devices (int)

• device_names (tuple[str, ...])

device_names: tuple[str, ...] = ()

name: str

num_devices: int