feat: move ALL_SYSTEMS

wozeparrot · Apr 30, 2024 · 7535a5a · 7535a5a
1 parent 1dff76c
commit 7535a5a
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/tinymod/common/benchmarks.py b/tinymod/common/benchmarks.py
@@ -26,6 +26,8 @@ def get_benchmarks(filename: str, system: str):
   "resnet": re.compile(r"\d+[ ]+(\d+\.\d+) ms run,"),
 }
 
+ALL_SYSTEMS = ["amd", "amd-train", "nvidia", "mac"]
+
 # regex, systems, skip_count, max_count
 TRACKED_BENCHMARKS = {
   # stable diffusion

diff --git a/tinymod/plugins/benchmarks.py b/tinymod/plugins/benchmarks.py
@@ -7,7 +7,7 @@
 
 import os, logging
 
-from common.benchmarks import REPO, BENCHMARKS_DIR, TRACKED_BENCHMARKS, CachedBenchmarks, filter_points
+from common.benchmarks import REPO, BENCHMARKS_DIR, TRACKED_BENCHMARKS, ALL_SYSTEMS, CachedBenchmarks, filter_points
 
 TinyMod: Client
 GUILD: Guild
@@ -24,7 +24,6 @@
 }
 CI_CHANNEL_ID = 1068993556905218128
 GITHUB_WEBHOOK_ID = 1068993579520884826
-ALL_SYSTEMS = ["amd", "amd-train", "mac", "nvidia"]
 RANGE = range(10, 100 + 1, 10)
 
 # ***** Downloading benchmarks *****