For the complete documentation index, see llms.txt. Markdown versions of all pages are available by appending .md to any URL (e.g. /get-started.md).
Python function
as_device_mapping
as_device_mapping()
max.experimental.sharding.as_device_mapping(device)
Coerces any way of naming where a value sits into a mapping.
-
Parameters:
-
device (Device | DeviceRef | DeviceMesh | DeviceMapping) – A single device, a mesh to replicate over, or a mapping.
-
Returns:
-
The equivalent
DeviceMapping. -
Raises:
-
TypeError – If
deviceis none of those. -
Return type:
-
DeviceMapping