code comments NFC

Author: rengolin

tools/lh-run

@@ -145,6 +145,7 @@ if __name__ == "__main__":
         print(optimized_module)
 
     # Initialize the device data
+    # TODO: Allow automatic inspection of the payload module if the user doesn't provide input shapes.
     buffers = []
     for shape in args.input_shape:
         for sub_shape in shape.split(","):
@@ -153,9 +154,8 @@ if __name__ == "__main__":
                 np.random.rand(*shape).astype(_get_input_type(args.input_type))
             )
 
-    # Create the runner and execute the module.
+    # Create the runner and execute/benchmark the module.
     runner = Runner(optimized_module, opt_level=args.O)
-
     if args.benchmark:
         time_array = runner.benchmark(host_input_buffers=buffers)
         print(f"{len(time_array)} runs: {np.mean(time_array)} seconds")
@@ -165,6 +165,7 @@ if __name__ == "__main__":
             host_input_buffers=buffers,
         )
 
+    # Optionally print the output tensor after execution.
     if args.print_tensor > 0:
         idx = args.print_tensor - 1
         print(f"Output: {buffers[idx]}")