Restructure include paths

This commit is contained in:
2025-11-23 20:57:08 +01:00
parent 4161caf3e1
commit 38cb0c9ac0
13 changed files with 15 additions and 15 deletions

View File

@@ -0,0 +1,28 @@
#pragma once
#include <cuda_runtime.h>
#include "layer.hpp"
namespace CUDANet::Kernels {
__global__ void max_pool(
const float* __restrict__ d_input,
float* __restrict__ d_output,
const Shape input_shape,
const Shape output_shape,
const Shape pool_shape,
const Shape stride_shape,
const Shape padding_shape
);
__global__ void avg_pool(
const float* __restrict__ d_input,
float* __restrict__ d_output,
const Shape input_shape,
const Shape output_shape,
const Shape pool_shape,
const Shape stride_shape,
const Shape padding_shape
);
} // namespace CUDANet::Kernels