Add main include file

This commit is contained in:
2024-05-19 16:16:14 +02:00
parent 4a1c4a5f91
commit 8bb2562c4c

32
include/cudanet.cuh Normal file
View File

@@ -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