fix thread block size

PaddlePaddle · Aug 11, 2022 · 50bf7da · 50bf7da
1 parent 36d1eab
commit 50bf7da
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/paddle/phi/kernels/gpu/graph_send_ue_recv_funcs.h b/paddle/phi/kernels/gpu/graph_send_ue_recv_funcs.h
@@ -54,8 +54,8 @@ inline int FindNumThreads(int dim, int max_num_threads) {
                     0,
                     phi::errors::PreconditionNotMet(
                         "Required dim >= 0, but received dim = %d", dim));
-  if (dim == 0) return 1;
   int res = max_num_threads;
+  if (dim == 0) res = 1;
   while (res > dim) {
     res = res >> 1;
   }