cubek-convolution 0.3.0

CubeK: Convolution Kernels
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22

mod algorithm;
mod precision;
mod tiling_scheme;

#[macro_export]
macro_rules! testgen_convolution_accelerated {
    () => {
        mod conv2d_accelerated {
            use super::*;

            mod cmma {
                use super::*;
                $crate::testgen_convolution_accelerated_algorithm!();
            }

            mod mma {
                use super::*;
                $crate::testgen_convolution_accelerated_algorithm!();
            }
        }
    };
}