Fix Legion control replication code.

StanfordLegion · Oct 5, 2023 · 5365a63 · 5365a63
1 parent 15a7d9d
commit 5365a63
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/legion/main.cc b/legion/main.cc
@@ -603,8 +603,7 @@ LegionApp::LegionApp(Runtime *runtime, Context ctx)
 
 void LegionApp::run()
 {
-  // FIXME (Elliott): Do this correctly for control replication
-  if (runtime->get_executing_processor(ctx).address_space() == 0) {
+  if (runtime->get_shard_id(ctx, true) == 0) {
     display();
   }
 
@@ -623,7 +622,7 @@ void LegionApp::run()
   unsigned long long stop = Realm::Clock::current_time_in_nanoseconds();
 
   double elapsed = (stop - start) / 1e9;
-  if (runtime->get_executing_processor(ctx).address_space() == 0) {
+  if (runtime->get_shard_id(ctx, true) == 0) {
     report_timing(elapsed);
   }
 }