fix some warning

7 years ago · 717e125213
parent ba4322f4dd
commit 717e125213
1 changed files with 1 additions and 1 deletions
--- a/paddle/platform/for_range.h
+++ b/paddle/platform/for_range.h
@ -62,7 +62,7 @@ struct ForRange<CUDADeviceContext> {

  template <typename Function>
  inline void operator()(Function func) const {
-    constexpr size_t num_threads = 1024;
+    constexpr int num_threads = 1024;
    int block_size = limit_ <= num_threads ? limit_ : num_threads;
    int grid_size = (limit_ + num_threads - 1) / num_threads;