Another compile fix

guanyuwang0001 · Aug 5, 2020 · 14b44a8 · 14b44a8
1 parent 737c355
commit 14b44a8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/softmax_layer.c b/src/softmax_layer.c
@@ -422,9 +422,9 @@ void forward_contrastive_layer(contrastive_layer l, network_state state)
  }
  int *labels = NULL;
  if (contr_size > 2) {
- cuda_push_array((contrastive_params *)l.contrast_p_gpu, (contrastive_params *)contrast_p, contr_size * sizeof(contrastive_params) / 4);
+ cuda_push_array((float *)l.contrast_p_gpu, (float *)contrast_p, contr_size * sizeof(contrastive_params) / 4);
  P_constrastive_f_det_gpu(labels, l.embedding_size, l.temperature, l.contrast_p_gpu, contr_size);
- cuda_pull_array((contrastive_params *)l.contrast_p_gpu, (contrastive_params *)contrast_p, contr_size * sizeof(contrastive_params) / 4);
+ cuda_pull_array((float *)l.contrast_p_gpu, (float *)contrast_p, contr_size * sizeof(contrastive_params) / 4);
  }
 #else // GPU
  int k;