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
use super::TypeKey;
use crate::ir::Ident;
use smallvec::SmallVec;
#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash, PartialOrd, Ord)]
pub(crate) struct ExprId(pub u32);
#[derive(Debug, Clone, PartialEq, Eq, Hash, PartialOrd, Ord)]
pub(crate) enum ExprKey {
LitU32(u32),
LitI32(i32),
/// Stores the IEEE 754 bits so `Eq`/`Hash`/`Ord` work correctly.
LitF32(u32),
LitBool(bool),
Var(Ident),
Load(Ident, ExprId),
BufLen(Ident),
InvocationId(u8),
WorkgroupId(u8),
LocalId(u8),
BinOp(u8, ExprId, ExprId),
UnOp(u8, ExprId),
/// CSE key for `BinOp::Opaque(id)` — stores the extension u32 id so
/// two extensions with distinct ids hash to distinct keys. Without
/// this, every `BinOp::Opaque(_)` collapsed onto a single key via
/// `bin_op_key`'s wildcard fallback — silently merging unrelated
/// extensions in CSE. (§1 injectivity contract.)
BinOpOpaque(u32, ExprId, ExprId),
/// CSE key for `UnOp::Opaque(id)` — same rationale as `BinOpOpaque`.
UnOpOpaque(u32, ExprId),
Call(Ident, SmallVec<[ExprId; 4]>),
Fma(ExprId, ExprId, ExprId),
Select(ExprId, ExprId, ExprId),
Cast(TypeKey, ExprId),
Atomic,
/// Subgroup intrinsics are effectful + lane-correlated; CSE must not
/// merge them, so every instance gets a unique counter-keyed identity.
Subgroup(u32),
SubgroupLocalId,
SubgroupSize,
Opaque(&'static str, [u8; 32]),
}