cubecl-cpu 0.10.0-pre.3

CPU runtime for CubeCL
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
use cubecl_core::ir::Instruction;

use super::Visitor;

impl<'a> Visitor<'a> {
    pub fn visit_instruction(&mut self, instruction: &Instruction) {
        match instruction.out {
            Some(out) => {
                self.visit_operation_with_out(&instruction.operation, out);
            }
            None => {
                self.visit_operation(&instruction.operation);
            }
        }
    }
}