Reduce log volume by moving output code logging behind HELION_PRINT_OUTPUT_CODE=1 (#1233)

Author: yf225

helion/runtime/kernel.py

@@ -528,10 +528,10 @@ def compile_config(
             self.maybe_log_repro(log.warning, self.fake_args, config=config)
             raise
         if allow_print:
-            log.info("Output code: \n%s", triton_code)
             log.info("Output code written to: %s", module.__file__)
             log.debug("Debug string: \n%s", LazyString(lambda: self._debug_str()))
             if self.settings.print_output_code:
+                log.info("Output code: \n%s", triton_code)
                 print(triton_code, file=sys.stderr)
         rv = getattr(module, self.kernel.name)
         self._compile_cache[config] = rv

test/test_logging.py

@@ -89,6 +89,7 @@ def add(x, y):
 
         x = torch.randn(4, device=DEVICE)
 
+        add.settings.print_output_code = True
         with self.assertLogs("helion.runtime.kernel", level="DEBUG") as cm:
             add(x, x)
         self.assertTrue(