Update parallel.h

secretflow · Dec 1, 2023 · 7181d24 · 7181d24
1 parent 0c6fd42
commit 7181d24
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/yacl/utils/parallel.h b/yacl/utils/parallel.h
@@ -51,6 +51,11 @@ body of your function, only data pointers.
 inline void parallel_for(int64_t begin, int64_t end, int64_t grain_size,
                          const std::function<void(int64_t, int64_t)>& f);
 
+inline void parallel_for(int64_t begin, int64_t end,
+                         const std::function<void(int64_t, int64_t)>& f) {
+  parallel_for(begin, end, 1, f);
+}
+
 /*
 parallel_reduce