mirror of
https://github.com/lordmathis/CUDANet.git
synced 2025-12-22 14:24:22 +00:00
Update main include file
This commit is contained in:
11
include/backend/cuda/cuda_backend.cuh
Normal file
11
include/backend/cuda/cuda_backend.cuh
Normal file
@@ -0,0 +1,11 @@
|
|||||||
|
#pragma once
|
||||||
|
|
||||||
|
// CUDA Backend Implementation
|
||||||
|
#include "backend/cuda/cuda.cuh"
|
||||||
|
|
||||||
|
// CUDA Kernels
|
||||||
|
#include "backend/cuda/kernels/activation_functions.cuh"
|
||||||
|
#include "backend/cuda/kernels/convolution.cuh"
|
||||||
|
#include "backend/cuda/kernels/matmul.cuh"
|
||||||
|
#include "backend/cuda/kernels/pool.cuh"
|
||||||
|
|
||||||
@@ -1,35 +1,55 @@
|
|||||||
#ifndef CUDANET_H
|
#pragma once
|
||||||
#define CUDANET_H
|
|
||||||
|
|
||||||
#ifdef USE_CUDA
|
// ============================================================================
|
||||||
#include "activation_functions.cuh"
|
// Core Data Structures & Abstractions (BACKEND-INDEPENDENT)
|
||||||
#include "convolution.cuh"
|
// ============================================================================
|
||||||
#include "matmul.cuh"
|
|
||||||
#include "pooling.cuh"
|
|
||||||
#endif
|
|
||||||
|
|
||||||
// Layers
|
#include "shape.hpp"
|
||||||
#include "activation.hpp"
|
#include "backend.hpp"
|
||||||
#include "add.hpp"
|
#include "tensor.hpp"
|
||||||
#include "avg_pooling.hpp"
|
|
||||||
#include "batch_norm.hpp"
|
|
||||||
#include "concat.hpp"
|
|
||||||
#include "conv2d.hpp"
|
|
||||||
#include "dense.hpp"
|
|
||||||
#include "input.hpp"
|
|
||||||
#include "layer.hpp"
|
#include "layer.hpp"
|
||||||
#include "max_pooling.hpp"
|
|
||||||
#include "output.hpp"
|
|
||||||
|
|
||||||
// Models
|
// ============================================================================
|
||||||
#include "model.hpp"
|
// Container Classes
|
||||||
|
// ============================================================================
|
||||||
|
|
||||||
#include "module.hpp"
|
#include "module.hpp"
|
||||||
|
#include "model.hpp"
|
||||||
|
|
||||||
|
// ============================================================================
|
||||||
|
// Layer Implementations
|
||||||
|
// ============================================================================
|
||||||
|
|
||||||
|
// Activation
|
||||||
|
#include "layers/activation.hpp"
|
||||||
|
|
||||||
|
// Normalization
|
||||||
|
#include "layers/batch_norm.hpp"
|
||||||
|
|
||||||
|
// Linear
|
||||||
|
#include "layers/dense.hpp"
|
||||||
|
|
||||||
|
// Convolutional
|
||||||
|
#include "layers/conv2d.hpp"
|
||||||
|
|
||||||
|
// Pooling
|
||||||
|
#include "layers/max_pool.hpp"
|
||||||
|
#include "layers/avg_pool.hpp"
|
||||||
|
|
||||||
|
// Composition (element-wise operations)
|
||||||
|
#include "layers/add.hpp"
|
||||||
|
#include "layers/concat.hpp"
|
||||||
|
|
||||||
|
// ============================================================================
|
||||||
|
// Utilities
|
||||||
|
// ============================================================================
|
||||||
|
|
||||||
|
#include "utils/imagenet.hpp"
|
||||||
|
|
||||||
|
// ============================================================================
|
||||||
|
// Backend-Specific Includes (conditionally compiled)
|
||||||
|
// ============================================================================
|
||||||
|
|
||||||
// Utils
|
|
||||||
#include "imagenet.hpp"
|
|
||||||
#ifdef USE_CUDA
|
#ifdef USE_CUDA
|
||||||
#include "cuda_helper.cuh"
|
#include "backend/cuda/cuda_backend.cuh"
|
||||||
#include "vector.cuh"
|
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
#endif // CUDANET_H
|
|
||||||
Reference in New Issue
Block a user