Mathis/CUDANet
1
0
Fork 0
mirror of https://github.com/lordmathis/CUDANet.git synced 2025-12-24 07:14:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix some dense layer issues

Browse Source
This commit is contained in:
Matúš Námešný
2025-11-18 22:17:08 +01:00
parent 7f203b8947
commit 4c26efe826
8 changed files with 110 additions and 44 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
include/backend.hpp
View File

@@ -32,12 +32,12 @@ class Backend {
) = 0;
virtual CUDANet::Tensor& dense(
CUDANet::Tensor& weights,
CUDANet::Tensor& biases,
CUDANet::Tensor& input,
const CUDANet::Tensor& weights,
const CUDANet::Tensor& biases,
const CUDANet::Tensor& input,
CUDANet::Tensor& output,
size_t input_size,
size_t output_size
const size_t input_size,
const size_t output_size
) = 0;
};

10
include/backend/cuda.cuh
View File

@@ -29,12 +29,12 @@ class CUDA : public Backend {
) override;
CUDANet::Tensor& dense(
CUDANet::Tensor& weights,
CUDANet::Tensor& biases,
CUDANet::Tensor& input,
const CUDANet::Tensor& weights,
const CUDANet::Tensor& biases,
const CUDANet::Tensor& input,
CUDANet::Tensor& output,
size_t input_size,
size_t output_size
const size_t input_size,
const size_t output_size
) override;
};

2
include/layer.hpp
View File

@@ -20,7 +20,7 @@ class Layer {
virtual ~Layer(){};
virtual CUDANet::Tensor& forward(CUDANet::Tensor &input) = 0;
virtual CUDANet::Tensor& forward(const CUDANet::Tensor &input) = 0;
virtual CUDANet::Shape input_shape() = 0;

2
include/layers/dense.hpp
View File

@@ -18,7 +18,7 @@ class Dense : public Layer {
~Dense();
CUDANet::Tensor& forward(CUDANet::Tensor &input) override;
CUDANet::Tensor& forward(const CUDANet::Tensor &input) override;
CUDANet::Shape input_shape() override;

6
include/tensor.hpp
View File

@@ -22,6 +22,12 @@ public:
Tensor() = default;
Tensor(Shape shape, DType dtype, CUDANet::Backend* backend);
Tensor(Tensor&& other) noexcept;
Tensor& operator=(Tensor&& other) noexcept;
Tensor(const Tensor&) = delete;
Tensor& operator=(const Tensor&) = delete;
~Tensor();
size_t size() const;