knok-macros 0.1.0

Procedural macros for knok
1
2
3
4
5
6
7
8
9
10
11

use proc_macro::TokenStream;

#[proc_macro_attribute]
pub fn graph(attr: TokenStream, item: TokenStream) -> TokenStream {
    knok_compile::expand_graph(attr.into(), item.into()).into()
}

#[proc_macro]
pub fn mlir_model(input: TokenStream) -> TokenStream {
    knok_compile::expand_mlir_model(input.into()).into()
}