From 311b2be0adb66763a79538dff211c5ae91cd21fe Mon Sep 17 00:00:00 2001
From: Iain Moncrief <moncrief@oregonstate.edu>
Date: Tue, 17 Sep 2024 11:19:01 -0700
Subject: [PATCH] Update MultiLocaleInference.chpl to have better comments

---
 examples/MultiLocaleInference.chpl | 14 ++++++--------
 1 file changed, 6 insertions(+), 8 deletions(-)

diff --git a/examples/MultiLocaleInference.chpl b/examples/MultiLocaleInference.chpl
index 8bba0bbac..ad9de97c2 100644
--- a/examples/MultiLocaleInference.chpl
+++ b/examples/MultiLocaleInference.chpl
@@ -38,23 +38,21 @@ var preds: [imagesD] int;
 config const numTries = 1;
 
 var totalTime: real;
-
 for i in 0..<numTries {
 
+    // Begin timing the inference on entire batch
     var st = new Time.stopwatch();
     st.start();
 
-    // coforall loc in Locales {
-    //     on loc {
-    //         const myAD = A.domain.localIndices();
-    //         forall i in myAD;
-    //     }
-    // }
+    // Feed each input through the network in parallel
     forall (image,pred) in zip(images,preds) {
+        // Get reference to the model instance on the iteration's node
         var model = localeModels[here.id].borrow();
+        // Map the input image to an int from 0-9
         pred = model(image).argmax();
     }
-    
+
+    // Stop the timer
     st.stop();
     const tm = st.elapsed();
     totalTime += tm;