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

Change unsigned int to int

Browse Source
This commit is contained in:
Matúš Námešný
2024-03-18 19:40:00 +01:00
parent aac0c3a826
commit e6d3757312
3 changed files with 19 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
include/kernels/convolution.cuh
View File

@@ -39,12 +39,13 @@ __global__ void convolution(
const float* __restrict__ d_input,
const float* __restrict__ d_kernel,
float* __restrict__ d_output,
const unsigned int inputSize,
const unsigned int nChannels,
const unsigned int kernelSize,
const unsigned int stride,
const unsigned int nFilters,
const unsigned int outputSize
const int inputSize,
const int nChannels,
const int paddingSize,
const int kernelSize,
const int stride,
const int nFilters,
const int outputSize
);
} // namespace CUDANet::Kernels

19
src/kernels/convolution.cu
View File

@@ -83,12 +83,13 @@ __global__ void CUDANet::Kernels::convolution(
const float* __restrict__ d_input,
const float* __restrict__ d_kernel,
float* __restrict__ d_output,
const unsigned int inputSize,
const unsigned int nChannels,
const unsigned int kernelSize,
const unsigned int stride,
const unsigned int nFilters,
const unsigned int outputSize
const int inputSize,
const int nChannels,
const int paddingSize,
const int kernelSize,
const int stride,
const int nFilters,
const int outputSize
) {
int tid = blockDim.x * blockIdx.x + threadIdx.x;
@@ -104,9 +105,9 @@ __global__ void CUDANet::Kernels::convolution(
float sum = 0.0f;
// 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++) {
for (int c = 0; c < nChannels; c++) {
for (int k = 0; k < kernelSize; k++) {
for (int l = 0; l < kernelSize; l++) {
int kernelIndex = f * kernelSize * kernelSize * nChannels +
c * kernelSize * kernelSize + k * kernelSize +
l;

4
src/layers/conv2d.cu
View File

@@ -124,8 +124,8 @@ float* Layers::Conv2d::forward(const float* d_input) {
// Convolve
THREADS_PER_BLOCK = outputSize * outputSize * numFilters;
Kernels::convolution<<<1, THREADS_PER_BLOCK>>>(
d_padded, d_weights, d_output, inputSize + (2 * paddingSize),
inputChannels, kernelSize, stride, numFilters, outputSize
d_padded, d_weights, d_output, inputSize + 2 * paddingSize, inputChannels, paddingSize,
kernelSize, stride, numFilters, outputSize
);
// Add bias