fix dtype of output error, should be int64_t

PaddlePaddle · Sep 29, 2020 · cf96fde · cf96fde · paddle-bot-old · Sep 29, 2020
1 parent b5b9903
commit cf96fde
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/paddle/fluid/operators/multinomial_op.cu b/paddle/fluid/operators/multinomial_op.cu
@@ -151,7 +151,7 @@ class MultinomialOpKernel<platform::CUDADeviceContext, T>
 
       MultinomialFunctor<T>(cpu_out_data, cpu_in_data, num_samples, replacement,
                             num_categories, num_distributions);
-      cudaMemcpy(out_data, cpu_out_data, out_data_numel * sizeof(T),
+      cudaMemcpy(out_data, cpu_out_data, out_data_numel * sizeof(int64_t),
                  cudaMemcpyHostToDevice);
 
       delete[] cpu_in_data;