cubecl-macros-internal 0.11.3

Internal procedural macros 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

use syn::{
    ExprClosure, Token, Type,
    parse::{Parse, ParseStream},
    punctuated::Punctuated,
    token,
};

pub struct Simplify {
    pub op_type: Type,
    _comma: Token![,],
    _brace: token::Brace,
    pub arms: Punctuated<ExprClosure, Token![,]>,
}

impl Parse for Simplify {
    fn parse(input: ParseStream) -> syn::Result<Self> {
        let content;
        Ok(Simplify {
            op_type: input.parse()?,
            _comma: input.parse()?,
            _brace: syn::braced!(content in input),
            arms: content.parse_terminated(ExprClosure::parse, Token![,])?,
        })
    }
}