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