add comment

exo-lang · Nov 4, 2022 · d0c5b50 · d0c5b50
1 parent 446b2cf
commit d0c5b50
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/exo/LoopIR_scheduling.py b/src/exo/LoopIR_scheduling.py
@@ -1079,6 +1079,10 @@ def calc_dim(self, dim):
             [w for w in self.win_stmt.rhs.idx if isinstance(w, LoopIR.Interval)]
         )
 
+        # Because our goal here is to offset `dim` in the original
+        # call argument to the point indexing to the windowing expression,
+        # new_dim should essencially be:
+        # `dim` + "number of LoopIR.Points in the windowing expression before the `dim` number of LoopIR.Interval"
         new_dim = 0
         for w in self.win_stmt.rhs.idx:
             if isinstance(w, LoopIR.Interval):