diff --git a/milabench/config.py b/milabench/config.py index 6f42dbf8..4936054d 100644 --- 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))