include system info in printed report

Author: ngc92

src/discord-cluster-manager/cogs/submit_cog.py

@@ -225,7 +225,7 @@ async def _handle_submission(
                 verdict = short_report[-1][0]
                 id_str = f"{verdict}" if submission_id == -1 else f"{verdict} #{submission_id}"
                 await reporter.generate_report(
-                    f"{id_str} {name} on {gpu_type.name} ({launcher.name})", result.runs
+                    f"{id_str} {name} on {gpu_type.name} ({launcher.name})", result
                 )
             except Exception as E:
                 logger.error("Error generating report. Result: %s", result, exc_info=E)

src/discord-cluster-manager/report.py

@@ -3,7 +3,7 @@
 
 import consts
 import discord
-from run_eval import CompileResult, EvalResult, RunResult
+from run_eval import CompileResult, EvalResult, FullResult, RunResult, SystemInfo
 from utils import format_time
 
 
@@ -250,6 +250,16 @@ def log_one(base_name):
         return "❗ Could not find any benchmarks"
 
 
+def generate_system_info(system: SystemInfo):
+    return f"""
+Running on:
+* GPU: `{system.gpu}`
+* CPU: `{system.cpu}`
+* Platform: `{system.platform}`
+* Torch: `{system.torch}`
+"""
+
+
 def generate_report(reporter: "RunResultReport", runs: dict[str, EvalResult]):  # noqa: C901
     if "test" in runs:
         test_run = runs["test"]
@@ -379,7 +389,7 @@ async def update_title(self, new_title):
     def get_message(self):
         return str.join("\n", [f"**{self.title}**"] + self.lines)
 
-    async def generate_report(self, title: str, runs: dict[str, EvalResult]):
+    async def generate_report(self, title: str, result: FullResult):
         raise NotImplementedError()
 
     async def _update_message(self):
@@ -400,15 +410,16 @@ def __init__(
     async def _update_message(self):
         await self.root._update_message()
 
-    async def generate_report(self, title: str, runs: dict[str, EvalResult]):
+    async def generate_report(self, title: str, result: FullResult):
         thread = await self.interaction.channel.create_thread(
             name=title,
             type=discord.ChannelType.private_thread,
             auto_archive_duration=1440,
         )
         await thread.add_user(self.interaction.user)
         report = RunResultReport()
-        generate_report(report, runs)
+        report.add_text(generate_system_info(result.system))
+        generate_report(report, result.runs)
         message = ""
         for part in report.data:
             if isinstance(part, Text):