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

use core::fmt::Display;

use crate::{CopyMemoryOperands, IndexOperands, StoreOperands, TypeHash, Value};

use crate::OperationReflect;

/// Bitwise operations
#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
#[derive(Debug, Clone, TypeHash, PartialEq, Eq, Hash, OperationReflect)]
#[operation(opcode_name = MemoryOpCode)]
pub enum Memory {
    #[operation(pure)]
    Index(IndexOperands),
    Load(#[args(allow_ptr, ptr_read)] Value),
    Store(StoreOperands),
    CopyMemory(CopyMemoryOperands),
}

impl Display for Memory {
    fn fmt(&self, f: &mut core::fmt::Formatter<'_>) -> core::fmt::Result {
        match self {
            Memory::Index(op) => write!(f, "&{}[{}]", op.list, op.index),
            Memory::Load(variable) => write!(f, "load({variable})"),
            Memory::Store(op) => write!(f, "store({}, {})", op.ptr, op.value),
            Memory::CopyMemory(op) => {
                write!(f, "memcpy({}, {}, {})", op.source, op.target, op.len)
            }
        }
    }
}