move comments

gussmith23 · Jan 20, 2021 · 1e7fdc5 · 1e7fdc5
1 parent e2baa41
commit 1e7fdc5
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/src/language/from_relay/mod.rs b/src/language/from_relay/mod.rs
@@ -676,6 +676,12 @@ fn compile_expression(
             .clone()
             .downcast::<tvm::ir::relay::Call>()
         {
+            // For batch norms, we use a hack where if
+            // simplify_batch_norm_for_inference_hack is set,
+            // we compile the Relay batch norm to a single output.
+
+            // All other expressions are compiled using Language::TupleGetItem construct
+
             if simplify_batch_norm_for_inference_hack
                 && call
                     .op
@@ -689,18 +695,13 @@ fn compile_expression(
                     == "nn.batch_norm"
                 && tuple_get_item.index == 0
             {
-                // For batch norms, we use a hack where if
-                // simplify_batch_norm_for_inference_hack is set,
-                // we compile the Relay batch norm to a single output.
                 get_compiled_expression(tuple_get_item.tuple.clone())
             } else {
-                // All other expressions are compiled using Language::TupleGetItem construct
                 let data_id = get_compiled_expression(tuple_get_item.tuple.clone());
                 let index_id = glenside_expr.add(Language::Usize(tuple_get_item.index as usize));
                 glenside_expr.add(Language::TupleGetItem([data_id, index_id]))
             }
         } else {
-            // All other expressions are compiled using Language::TupleGetItem construct
             let data_id = get_compiled_expression(tuple_get_item.tuple.clone());
             let index_id = glenside_expr.add(Language::Usize(tuple_get_item.index as usize));
             glenside_expr.add(Language::TupleGetItem([data_id, index_id]))