diff --git a/include/cudanet.cuh b/include/cudanet.cuh new file mode 100644 index 0000000..528aa45 --- /dev/null +++ b/include/cudanet.cuh @@ -0,0 +1,32 @@ +#ifndef CUDANET_H +#define CUDANET_H + +// Kernels +#include "activation_functions.cuh" +#include "convolution.cuh" +#include "matmul.cuh" +#include "pooling.cuh" + +// Layers +#include "activation.cuh" +#include "add.cuh" +#include "avg_pooling.cuh" +#include "batch_norm.cuh" +#include "concat.cuh" +#include "conv2d.cuh" +#include "dense.cuh" +#include "input.cuh" +#include "layer.cuh" +#include "max_pooling.cuh" +#include "output.cuh" + +// Models +#include "model.hpp" +#include "module.hpp" + +// Utils +#include "cuda_helper.cuh" +#include "imagenet.hpp" +#include "vector.cuh" + +#endif // CUDANET_H \ No newline at end of file