melior-macro 0.20.2

Internal macros for Melior
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

#[derive(Clone, Debug, Eq, PartialEq)]
pub enum VariadicKind {
    Simple {
        unfixed_seen: bool,
    },
    SameSize {
        unfixed_count: usize,
        preceding_simple_count: usize,
        preceding_variadic_count: usize,
    },
    AttributeSized,
    // TODO Support variadic-of-variadic operands.
    // https://mlir.llvm.org/docs/DefiningDialects/Operations/#variadicofvariadic-operands
}

impl VariadicKind {
    pub fn new(
        unfixed_count: usize,
        same_size: bool,
        attribute_sized: bool,
    ) -> Result<Self, &'static str> {
        if unfixed_count <= 1 {
            Ok(Self::Simple {
                unfixed_seen: false,
            })
        } else if same_size {
            Ok(Self::SameSize {
                unfixed_count,
                preceding_simple_count: 0,
                preceding_variadic_count: 0,
            })
        } else if attribute_sized {
            Ok(Self::AttributeSized)
        } else {
            // TODO: Support multiple variadic operands/results without these traits.
            Err(
                "multiple variadic operands/results require SameVariadicOperandSize, \
                SameVariadicResultSize, or AttrSizedOperandSegments/AttrSizedResultSegments trait",
            )
        }
    }
}