src/network.c
@@ -443,8 +443,8 @@ } #ifdef GPU if(gpu_index >= 0){ printf(" try to allocate workspace = %zu * sizeof(float), ", (workspace_size - 1) / sizeof(float) + 1); net->workspace = cuda_make_array(0, (workspace_size-1)/sizeof(float)+1); printf(" try to allocate workspace = %zu * sizeof(float), ", workspace_size / sizeof(float) + 1); net->workspace = cuda_make_array(0, workspace_size/sizeof(float) + 1); printf(" CUDA allocate done! \n"); }else { free(net->workspace);