Module auto_device

Module auto_device 

Source
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§

AutoDevice
A wrapper for arrays that automatically chooses the best device.
AutoDeviceConfig
Configuration for automatic device placement.

Enums§

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