cubecl-core 0.11.0-pre.1

CubeCL core create
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

#![allow(unknown_lints, unnecessary_transmutes)]

use alloc::{vec, vec::Vec};
use cubecl_ir::{GlobalState, Instruction, OperationReflect};

use crate::post_processing::{
    analysis_helper::GlobalAnalyses, util::AtomicCounter, visitor::InstructionVisitor,
};

/// Eliminate non-output variables that are never read in the program.
#[derive(Debug)]
pub struct EliminateUnusedExpressions;

impl InstructionVisitor for EliminateUnusedExpressions {
    fn visit_instruction(
        &mut self,
        instruction: Instruction,
        _global_state: &GlobalState,
        analyses: &GlobalAnalyses,
        changes: &AtomicCounter,
    ) -> Vec<Instruction> {
        if let Some(out) = instruction.out
            && instruction.operation.is_pure()
            && !analyses.used_values().contains(&out)
        {
            changes.inc();
            vec![]
        } else {
            vec![instruction]
        }
    }
}