mirror/darknet
1
0
Fork 0
mirror of https://github.com/pjreddie/darknet.git synced 2023-08-10 21:13:14 +03:00
Code Issues Projects Releases Wiki Activity

Compile fix

Browse Source
This commit is contained in:
AlexeyAB
2018-09-07 15:07:46 +03:00
parent 007878393f
commit b141f85cab
5 changed files with 164 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/convolutional_layer.c
View File

@ -637,6 +637,8 @@ void binary_align_weights(convolutional_layer *l)
check_error(status);
status = cudaMemcpy(l->align_bit_weights_gpu, l->align_bit_weights, l->align_bit_weights_size, cudaMemcpyHostToDevice);
check_error(status);
status = cudaMemcpy(l->binary_weights_gpu, l->binary_weights, m*k*sizeof(float), cudaMemcpyHostToDevice);
check_error(status);
l->mean_arr_gpu = cuda_make_array(l->mean_arr, l->n);
cudaDeviceSynchronize();