Mathis/CUDANet
1
0
Fork 0
mirror of https://github.com/lordmathis/CUDANet.git synced 2025-11-06 17:54:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

Make conv2d work again

Browse Source
This commit is contained in:
Matúš Námešný
2024-03-10 19:13:22 +01:00
parent 6bbc036f62
commit f3112311da
6 changed files with 146 additions and 98 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
src/kernels/convolution.cu
View File

@@ -1,4 +1,5 @@
#include "convolution.cuh"
#include <iostream>
__global__ void convolution_kernel(
const float* d_input,
@@ -19,35 +20,26 @@ __global__ void convolution_kernel(
// Get output index
int f = tid / (outputSize * outputSize);
int i = (tid % (outputSize * outputSize)) / outputSize;
int j = (tid % (outputSize * outputSize)) % outputSize;
int i = tid % (outputSize * outputSize) / outputSize;
int j = tid % outputSize;
float sum = 0.0f;
// std::cout << "f: " << f << ", i: " << i << ", j: " << j << std::endl;
// Iterate over kernel and input matrix
for (int k = 0; k < kernelSize; k++) {
for (int l = 0; l < kernelSize; l++) {
for (int c = 0; c < nChannels; c++) {
int kernelIndex =
k * (kernelSize * nChannels * nFilters) +
l * (nChannels * nFilters) + c * (nFilters) + f;
int inputIndex =
(i * stride + k) * (inputSize * nChannels) +
(j * stride + l) * (nChannels) + c;
// std::cout << "kernelIndex: " << kernelIndex << ", kernel
// value: " << kernels[kernelIndex] << ", inputIndex: " <<
// inputIndex << ", input value: " << input[inputIndex] <<
// std::endl;
int kernelIndex = f * kernelSize * kernelSize * nChannels +
c * kernelSize * kernelSize + k * kernelSize +
l;
int inputIndex = c * inputSize * inputSize +
(i * stride + k) * inputSize +
(j * stride + l);
sum += d_kernel[kernelIndex] * d_input[inputIndex];
}
}
}
// std::cout << "sum: " << sum << std::endl;
d_output[i * (outputSize * nFilters) + j * (nFilters) + f] = sum;
d_output[tid] = sum;
}