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

Remove not needed code

Browse Source
This commit is contained in:
Matúš Námešný
2024-02-29 22:21:32 +01:00
parent 0f0e57b819
commit 045359cca2
7 changed files with 14 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/layers/dense.cu
View File

@@ -81,6 +81,8 @@ void Layers::Dense::forward(const float* d_input, float* d_output) {
d_output, d_output, outputSize
);
}
CUDA_CHECK(cudaDeviceSynchronize());
}
void Layers::Dense::toCuda() {