Updates for Explicit policy in capturing benchmark information

utcs-scea · Jul 23, 2024 · d2b5c68 · d2b5c68
1 parent 2ffe91b
commit d2b5c68
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 4 deletions.
diff --git a/include/pando-lib-galois/loops/do_all.hpp b/include/pando-lib-galois/loops/do_all.hpp
@@ -597,10 +597,18 @@ template <typename State, typename R, typename F>
 pando::Status doAll(WaitGroup::HandleType wgh, State s, R range, const F& func) {
   return DoAll::doAll<State, R, F>(wgh, s, range, func);
 }
+template <SchedulerPolicy POLICY, typename State, typename R, typename F>
+pando::Status doAllExplicitPolicy(WaitGroup::HandleType wgh, State s, R range, const F& func) {
+  return DoAll::doAllExplicitPolicy<POLICY, State, R, F>(wgh, s, range, func);
+}
 template <typename R, typename F>
 pando::Status doAll(WaitGroup::HandleType wgh, R range, const F& func) {
   return DoAll::doAll<R, F>(wgh, range, func);
 }
+template <SchedulerPolicy POLICY, typename R, typename F>
+pando::Status doAllExplicitPolicy(WaitGroup::HandleType wgh, R range, const F& func) {
+  return DoAll::doAllExplicitPolicy<POLICY, R, F>(wgh, range, func);
+}
 template <typename State, typename R, typename F, typename L>
 pando::Status doAll(State s, R range, const F& func, const L& localityFunc) {
   return DoAll::doAll<State, R, F, L>(s, range, func, localityFunc);

diff --git a/microbench/bfs/include/pando-bfs-galois/sssp.hpp b/microbench/bfs/include/pando-bfs-galois/sssp.hpp
@@ -134,7 +134,7 @@ pando::Status SSSP_DLCSR(
   state.dist = 0;
 
 #ifdef PANDO_STAT_TRACE_ENABLE
-  PANDO_CHECK(galois::doAll(
+  PANDO_CHECK(galois::doAllExplicitPolicy<SchedulerPolicy::RANDOM>(
       wgh, phbfs, +[](pando::Vector<typename G::VertexTopologyID>) {
         PANDO_MEM_STAT_NEW_KERNEL("BFS Start");
       }));
@@ -168,7 +168,7 @@ pando::Status SSSP_DLCSR(
   }
 
 #ifdef PANDO_STAT_TRACE_ENABLE
-  PANDO_CHECK(galois::doAll(
+  PANDO_CHECK(galois::doAllExplicitPolicy<SchedulerPolicy::RANDOM>(
       wgh, phbfs, +[](pando::Vector<typename G::VertexTopologyID>) {
         PANDO_MEM_STAT_NEW_KERNEL("BFS END");
       }));
@@ -311,7 +311,7 @@ pando::Status SSSPMDLCSR(G& graph, std::uint64_t src, HostLocalStorage<MDWorkLis
 #endif
 
 #ifdef PANDO_STAT_TRACE_ENABLE
-  PANDO_CHECK(galois::doAll(
+  PANDO_CHECK(galois::doAllExplicitPolicy<SchedulerPolicy::RANDOM>(
       wgh, toRead, +[](MDWorkList<G>) {
         PANDO_MEM_STAT_NEW_KERNEL("BFS Start");
       }));
@@ -360,7 +360,7 @@ pando::Status SSSPMDLCSR(G& graph, std::uint64_t src, HostLocalStorage<MDWorkLis
   }
 
 #ifdef PANDO_STAT_TRACE_ENABLE
-  PANDO_CHECK(galois::doAll(
+  PANDO_CHECK(galois::doAllExplicitPolicy<SchedulerPolicy::RANDOM>(
       wgh, toRead, +[](MDWorkList<G>) {
         PANDO_MEM_STAT_NEW_KERNEL("BFS END");
       }));