Use number of specified thread for the compilation (#2978)

ROCm · Apr 25, 2024 · d41c3cf · d41c3cf
1 parent e5432cb
commit d41c3cf
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/include/migraphx/par_for.hpp b/src/include/migraphx/par_for.hpp
@@ -38,9 +38,9 @@ void par_for(std::size_t n, F f)
 }
 
 template <class F>
-void par_for(std::size_t n, std::size_t, F f)
+void par_for(std::size_t n, std::size_t min_grain, F f)
 {
-    par_for(n, f);
+    simple_par_for(n, min_grain, f);
 }
 
 } // namespace MIGRAPHX_INLINE_NS