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

Fix small layer issues

Browse Source
This commit is contained in:
Matúš Námešný
2025-11-22 00:33:51 +01:00
parent 4c8b2ef537
commit 104d6ea33d
6 changed files with 12 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/backends/cuda/layer_ops.cu
View File

@@ -98,7 +98,7 @@ CUDANet::Tensor& CUDA::conv2d(
dim3 grid(
(out_shape[0] + block.x - 1) / block.x,
(out_shape[1] + block.y - 1) / block.y,
(out_shape[3] + block.z - 1) / block.z
(out_shape[2] + block.z - 1) / block.z
);
Kernels::convolution<<<grid, block>>>(
@@ -212,6 +212,7 @@ CUDANet::Tensor& CUDA::batch_norm(
CUDA_CHECK(cudaGetLastError());
}
CUDA_CHECK(cudaDeviceSynchronize());
return output;
}
CUDANet::Tensor& CUDA::concat(