burn-std 0.21.0-pre.4

Core types and utilities shared across the Burn ecosystem.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43

use cubecl_common::config::logger::{LogLevel, LoggerConfig};

/// Configuration for autodiff in Burn.
#[derive(Default, Clone, Debug, serde::Serialize, serde::Deserialize)]
pub struct AutodiffConfig {
    /// Logger configuration for autodiff logs.
    #[serde(default)]
    pub logger: LoggerConfig<AutodiffLogLevel>,
}

/// Log levels for autodiff logging.
#[derive(
    Default,
    Clone,
    Copy,
    Debug,
    PartialEq,
    Eq,
    PartialOrd,
    Ord,
    serde::Serialize,
    serde::Deserialize,
)]
pub enum AutodiffLogLevel {
    /// Autodiff logging is disabled.
    #[default]
    #[serde(rename = "disabled")]
    Disabled,

    /// Log backward graph size and the checkpoint strategy applied per forward pass.
    #[serde(rename = "basic")]
    Basic,

    /// Additionally log each tensor that gets checkpointed or recomputed.
    #[serde(rename = "medium")]
    Medium,

    /// Log every graph node traversal and recomputation event.
    #[serde(rename = "full")]
    Full,
}

impl LogLevel for AutodiffLogLevel {}