Expand description
Automatic device placement for array operations.
This module provides functionality for automatically determining the best device (CPU, GPU, distributed) for array operations based on array size, available hardware, and operation complexity.
Modules§
- ops
- Implementation of common array operations with automatic device selection.
Structs§
- Auto
Device - A wrapper for arrays that automatically chooses the best device.
- Auto
Device Config - Configuration for automatic device placement.
Enums§
- Device
Type - Available device types for array operations.
Statics§
- AUTO_
DEVICE_ CONFIG - Global auto device configuration.
Functions§
- auto_
execute - Execute an operation with automatic device selection.
- convert_
to_ device - Convert an array to the specified device type.
- determine_
best_ device - Determine the best device for an array.
- determine_
best_ device_ for_ operation - Determine the best device for an operation with multiple arrays.
- get_
auto_ device_ config - Get the current auto device configuration.
- set_
auto_ device_ config - Set the global auto device configuration.