StepanovPlaton/NeuralNetwork
1
0
Fork 0
mirror of https://github.com/StepanovPlaton/NeuralNetwork.git synced 2026-04-03 20:30:39 +04:00
Code Issues Packages Projects Releases Wiki Activity

First NN forward

Browse Source
This commit is contained in:
Степанов Платон
2025-10-30 23:26:53 +04:00
parent 8d8335eca0
commit f728261354
10 changed files with 254 additions and 130 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
src/math/matrix/gpu/matrix.cpp
View File

@@ -1,15 +1,35 @@
#include <random>
#include "matrix.hpp"
std::random_device rd;
std::mt19937 gen(rd());
Matrices::GPU::GPU(int rows, int cols)
: IMatrix(rows, cols), queue(openCL.getContext(), openCL.getDevice()) {
validateDimensions(rows, cols);
std::vector<float> matrix;
matrix.reserve(rows * cols);
for (size_t i = 0; i < (size_t)rows * (size_t)cols; ++i)
matrix.push_back(std::generate_canonical<float, 32>(gen));
buffer = new cl::Buffer(openCL.getContext(), CL_MEM_READ_WRITE,
rows * cols * sizeof(float));
queue.enqueueWriteBuffer(*buffer, CL_TRUE, 0, rows * cols * sizeof(float),
matrix.data());
queue.finish();
}
Matrices::GPU::GPU(int rows, int cols, const std::vector<float> &matrix)
: IMatrix(rows, cols), queue(openCL.getContext(), openCL.getDevice()) {
validateDimensions(rows, cols);
if (matrix.size() != static_cast<size_t>(rows * cols)) {
throw std::invalid_argument("Matrix data size doesn't match dimensions");
}
buffer = new cl::Buffer(
openCL.getContext(), CL_MEM_READ_ONLY | CL_MEM_COPY_HOST_PTR,
rows * cols * sizeof(float), const_cast<float *>(matrix.data()));
buffer = new cl::Buffer(openCL.getContext(), CL_MEM_READ_WRITE,
rows * cols * sizeof(float));
queue.enqueueWriteBuffer(*buffer, CL_TRUE, 0, rows * cols * sizeof(float),
matrix.data());
queue.finish();
}
const std::vector<float> Matrices::GPU::toVector() const {