cubecl-ir 0.11.0-pre.1

Intermediate representation for CubeCL
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

use core::{cell::Ref, fmt::Display};

use alloc::string::ToString;
use alloc::vec::Vec;

use crate::{GlobalState, GlobalStateInner};

use super::Instruction;

pub trait Processor: core::fmt::Debug {
    fn transform(&self, processing: ScopeProcessing) -> ScopeProcessing;
}

/// Information necessary when compiling a scope.
pub struct ScopeProcessing {
    /// The operations.
    pub instructions: Vec<Instruction>,
    /// The global state
    pub global_state: GlobalState,
}

impl ScopeProcessing {
    pub fn state(&self) -> Ref<'_, GlobalStateInner> {
        self.global_state.borrow()
    }
}

impl Display for ScopeProcessing {
    fn fmt(&self, f: &mut core::fmt::Formatter<'_>) -> core::fmt::Result {
        writeln!(f, "{{")?;
        for instruction in self.instructions.iter() {
            let instruction_str = instruction.to_string();
            if !instruction_str.is_empty() {
                writeln!(f, "    {instruction_str}")?;
            }
        }
        write!(f, "}}")?;
        Ok(())
    }
}