Parse rocprofiler options

AMDResearch · Aug 2, 2024 · 377b1f3 · 377b1f3
1 parent 22900ec
commit 377b1f3
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/omnistat/monitor.py b/omnistat/monitor.py
@@ -58,6 +58,9 @@ def __init__(self, config):
         self.runtimeConfig["collector_enable_amd_smi_process"] = config["omnistat.collectors"].getboolean(
             "enable_amd_smi_process", False
         )
+        self.runtimeConfig["collector_enable_rocprofiler"] = config["omnistat.collectors"].getboolean(
+            "enable_rocprofiler", False
+        )
         self.runtimeConfig["collector_port"] = config["omnistat.collectors"].get("port", 8000)
         self.runtimeConfig["collector_usermode"] = config["omnistat.collectors"].getboolean("usermode", False)
         self.runtimeConfig["collector_rocm_path"] = config["omnistat.collectors"].get("rocm_path", "/opt/rocm")
@@ -83,6 +86,9 @@ def __init__(self, config):
             if config.has_option("omnistat.collectors.rms", "host_skip"):
                 self.runtimeConfig["rms_collector_host_skip"] = config["omnistat.collectors.rms"]["host_skip"]
 
+        if config.has_option("omniwatch.collectors.rocprofiler", "metrics"):
+            self.runtimeConfig["rocprofiler_metrics"] = config["omniwatch.collectors.rocprofiler"]["metrics"].split(",")
+
         # defined global prometheus metrics
         self.__globalMetrics = {}
         self.__registry_global = CollectorRegistry()