enki-anu 0.2.5

BorrowEngine, task recording, and pipeline synthesis for the Enki runtime
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
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48

use super::context::IngressContext;
use super::descriptor::ArgDescriptor;
use super::traits::{GpuType, NamArgs};

macro_rules! impl_nam_args {
    ($($T:ident),*) => {
        impl<$($T: GpuType),*> NamArgs for ($($T,)*) {
            #[allow(non_snake_case)]
            #[allow(unused_assignments)]
            fn collect_all<'a>(self) -> IngressContext<'a> {
                let mut ctx = IngressContext::new();
                let ($($T,)*) = self;

                let mut arg_idx = 0;
                $(
                    ctx.current_arg_index = arg_idx;
                    $T.collect(&mut ctx);
                    arg_idx += 1;
                )*

                ctx
            }

            fn describe_all() -> Vec<ArgDescriptor> {
                vec![
                    $(<$T as GpuType>::describe()),*
                ]
            }
        }
    };
}

impl_nam_args!(A);
impl_nam_args!(A, B);
impl_nam_args!(A, B, C);
impl_nam_args!(A, B, C, D);
impl_nam_args!(A, B, C, D, E);
impl_nam_args!(A, B, C, D, E, F);
impl_nam_args!(A, B, C, D, E, F, G);
impl_nam_args!(A, B, C, D, E, F, G, H);
impl_nam_args!(A, B, C, D, E, F, G, H, I);
impl_nam_args!(A, B, C, D, E, F, G, H, I, J);
impl_nam_args!(A, B, C, D, E, F, G, H, I, J, K);
impl_nam_args!(A, B, C, D, E, F, G, H, I, J, K, L);
impl_nam_args!(A, B, C, D, E, F, G, H, I, J, K, L, M);
impl_nam_args!(A, B, C, D, E, F, G, H, I, J, K, L, M, N);
impl_nam_args!(A, B, C, D, E, F, G, H, I, J, K, L, M, N, O);
impl_nam_args!(A, B, C, D, E, F, G, H, I, J, K, L, M, N, O, P);