Partial revert of monitor_tags

mila-iqia · Oct 7, 2024 · 13dd93d · 13dd93d
1 parent 65e2428
commit 13dd93d
Showing 1 changed file with 0 additions and 9 deletions.
diff --git a/milabench/config.py b/milabench/config.py
@@ -12,8 +12,6 @@
 config_global = contextvars.ContextVar("config", default=None)
 execution_count = (0, 0)
 
-_MONITOR_TAGS = {"monogpu", "multigpu", "multinode"}
-
 
 def set_run_count(total_run, total_bench):
     global execution_count
@@ -93,13 +91,6 @@ def finalize_config(name, bench_config):
             pack = (XPath(bench_config["config_base"]) / pack).resolve()
             bench_config["definition"] = str(pack)
 
-    if not name.startswith("_") and name != "*":
-        _tags = set(bench_config["tags"])
-        _monitor_tags = _tags & _MONITOR_TAGS
-        assert len(_monitor_tags) == 1, (
-            f"Bench {name} should have exactly one monitor tag. Found {_monitor_tags}"
-        )
-
     bench_config["tag"] = [bench_config["name"]]
 
     bench_config = OmegaConf.to_object(OmegaConf.create(bench_config))