List of all items

Structs

• device::DeviceConfig
• logging::LogConfig
• memory::MemoryTracker

Enums

• dtype::PrecisionMode
• error::CoreError
• logging::LogLevel

Traits

• dtype::DTypeExt
• traits::Dequantize
• traits::GpuDispatchable
• traits::Quantize
• traits::ValidatableConfig

Macros

• log_metric
• logging::debug
• logging::error
• logging::info
• logging::trace
• logging::warn

Functions

• device::get_device
• device::warn_if_cpu
• dtype::bytes_per_element
• dtype::is_floating_point
• logging::init_logging
• logging::log_memory_usage
• logging::log_training_step
• memory::estimate_attention_memory
• memory::estimate_tensor_bytes

Type Aliases

• error::Result

Constants

• VERSION
• memory::DEFAULT_OVERHEAD_FACTOR