amd-node-scraper 0.0.1__py3-none-any.whl

nodescraper/plugins/inband/pcie/pcie_plugin.py

@@ -0,0 +1,43 @@
+###############################################################################
+#
+# MIT License
+#
+# Copyright (c) 2025 Advanced Micro Devices, Inc.
+#
+# Permission is hereby granted, free of charge, to any person obtaining a copy
+# of this software and associated documentation files (the "Software"), to deal
+# in the Software without restriction, including without limitation the rights
+# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+# copies of the Software, and to permit persons to whom the Software is
+# furnished to do so, subject to the following conditions:
+#
+# The above copyright notice and this permission notice shall be included in all
+# copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+# SOFTWARE.
+#
+###############################################################################
+from nodescraper.base import InBandDataPlugin
+
+from .analyzer_args import PcieAnalyzerArgs
+from .pcie_analyzer import PcieAnalyzer
+from .pcie_collector import PcieCollector
+from .pcie_data import PcieDataModel
+
+
+class PciePlugin(InBandDataPlugin[PcieDataModel, None, PcieAnalyzerArgs]):
+    """Plugin for collection and analysis of PCIe data"""
+
+    DATA_MODEL = PcieDataModel
+
+    COLLECTOR = PcieCollector
+
+    ANALYZER = PcieAnalyzer
+
+    ANALYZER_ARGS = PcieAnalyzerArgs

nodescraper/plugins/inband/process/__init__.py

@@ -0,0 +1,25 @@
+###############################################################################
+#
+# MIT License
+#
+# Copyright (c) 2025 Advanced Micro Devices, Inc.
+#
+# Permission is hereby granted, free of charge, to any person obtaining a copy
+# of this software and associated documentation files (the "Software"), to deal
+# in the Software without restriction, including without limitation the rights
+# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+# copies of the Software, and to permit persons to whom the Software is
+# furnished to do so, subject to the following conditions:
+#
+# The above copyright notice and this permission notice shall be included in all
+# copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+# SOFTWARE.
+#
+###############################################################################

nodescraper/plugins/inband/process/analyzer_args.py

@@ -0,0 +1,45 @@
+###############################################################################
+#
+# MIT License
+#
+# Copyright (c) 2025 Advanced Micro Devices, Inc.
+#
+# Permission is hereby granted, free of charge, to any person obtaining a copy
+# of this software and associated documentation files (the "Software"), to deal
+# in the Software without restriction, including without limitation the rights
+# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+# copies of the Software, and to permit persons to whom the Software is
+# furnished to do so, subject to the following conditions:
+#
+# The above copyright notice and this permission notice shall be included in all
+# copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+# SOFTWARE.
+#
+###############################################################################
+
+from nodescraper.models import AnalyzerArgs
+from nodescraper.plugins.inband.process.processdata import ProcessDataModel
+
+
+class ProcessAnalyzerArgs(AnalyzerArgs):
+    max_kfd_processes: int = 0
+    max_cpu_usage: float = 20.0
+
+    @classmethod
+    def build_from_model(cls, datamodel: ProcessDataModel) -> "ProcessAnalyzerArgs":
+        """build analyzer args from data model
+
+        Args:
+            datamodel (ProcessDataModel): data model for plugin
+
+        Returns:
+            ProcessAnalyzerArgs: instance of analyzer args class
+        """
+        return cls(max_kfd_processes=datamodel.kfd_process, max_cpu_usage=datamodel.cpu_usage)

nodescraper/plugins/inband/process/collector_args.py

@@ -0,0 +1,31 @@
+###############################################################################
+#
+# MIT License
+#
+# Copyright (c) 2025 Advanced Micro Devices, Inc.
+#
+# Permission is hereby granted, free of charge, to any person obtaining a copy
+# of this software and associated documentation files (the "Software"), to deal
+# in the Software without restriction, including without limitation the rights
+# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+# copies of the Software, and to permit persons to whom the Software is
+# furnished to do so, subject to the following conditions:
+#
+# The above copyright notice and this permission notice shall be included in all
+# copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+# SOFTWARE.
+#
+###############################################################################
+
+from nodescraper.models import CollectorArgs
+
+
+class ProcessCollectorArgs(CollectorArgs):
+    top_n_process: int = 10

nodescraper/plugins/inband/process/process_analyzer.py

@@ -0,0 +1,91 @@
+###############################################################################
+#
+# MIT License
+#
+# Copyright (c) 2025 Advanced Micro Devices, Inc.
+#
+# Permission is hereby granted, free of charge, to any person obtaining a copy
+# of this software and associated documentation files (the "Software"), to deal
+# in the Software without restriction, including without limitation the rights
+# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+# copies of the Software, and to permit persons to whom the Software is
+# furnished to do so, subject to the following conditions:
+#
+# The above copyright notice and this permission notice shall be included in all
+# copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+# SOFTWARE.
+#
+###############################################################################
+from typing import Optional
+
+from nodescraper.enums import EventCategory, EventPriority, ExecutionStatus
+from nodescraper.interfaces import DataAnalyzer
+from nodescraper.models import TaskResult
+
+from .analyzer_args import ProcessAnalyzerArgs
+from .processdata import ProcessDataModel
+
+
+class ProcessAnalyzer(DataAnalyzer[ProcessDataModel, ProcessAnalyzerArgs]):
+    """Check cpu and kfd processes are within allowed maximum cpu and gpu usage"""
+
+    DATA_MODEL = ProcessDataModel
+
+    def analyze_data(
+        self, data: ProcessDataModel, args: Optional[ProcessAnalyzerArgs] = None
+    ) -> TaskResult:
+        """
+        Analyze the process data to check if the number of KFD processes and CPU usage
+        are within the allowed limits.
+
+        Args:
+            data (ProcessDataModel): The process data to analyze.
+            args (Optional[ProcessAnalyzerArgs], optional): The process analysis arguments. Defaults to None.
+
+        Returns:
+            TaskResult: The result of the analysis, containing any events logged during the process.
+        """
+        if not args:
+            args = ProcessAnalyzerArgs()
+
+        err_messages = []
+        if data.kfd_process is not None and data.kfd_process > args.max_kfd_processes:
+            err_messages.append(
+                f"Kfd processes {data.kfd_process} exeed max limit {args.max_kfd_processes}"
+            )
+            self._log_event(
+                category=EventCategory.OS,
+                description="Kfd processes exceed maximum limit",
+                data={
+                    "kfd_process": data.kfd_process,
+                    "kfd_process_limit": args.max_kfd_processes,
+                },
+                priority=EventPriority.CRITICAL,
+                console_log=True,
+            )
+
+        if data.cpu_usage is not None and data.cpu_usage > args.max_cpu_usage:
+            err_messages.append(f"CPU usage {data.cpu_usage} exceeds limit {args.max_cpu_usage}")
+            self._log_event(
+                category=EventCategory.OS,
+                description="CPU usage exceeds maximum limit",
+                data={
+                    "cpu_usage": data.cpu_usage,
+                    "cpu_usage_limit": args.max_cpu_usage,
+                },
+                priority=EventPriority.CRITICAL,
+                console_log=True,
+            )
+
+        if err_messages:
+            self.result.status = ExecutionStatus.ERROR
+            self.result.message = ". ".join(err_messages)
+
+        return self.result

nodescraper/plugins/inband/process/process_collector.py

@@ -0,0 +1,115 @@
+###############################################################################
+#
+# MIT License
+#
+# Copyright (c) 2025 Advanced Micro Devices, Inc.
+#
+# Permission is hereby granted, free of charge, to any person obtaining a copy
+# of this software and associated documentation files (the "Software"), to deal
+# in the Software without restriction, including without limitation the rights
+# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+# copies of the Software, and to permit persons to whom the Software is
+# furnished to do so, subject to the following conditions:
+#
+# The above copyright notice and this permission notice shall be included in all
+# copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+# SOFTWARE.
+#
+###############################################################################
+import re
+from typing import Optional
+
+from nodescraper.base import InBandDataCollector
+from nodescraper.enums import EventCategory, EventPriority, ExecutionStatus, OSFamily
+from nodescraper.models import TaskResult
+
+from .collector_args import ProcessCollectorArgs
+from .processdata import ProcessDataModel
+
+
+class ProcessCollector(InBandDataCollector[ProcessDataModel, ProcessCollectorArgs]):
+    """Collect Process details"""
+
+    SUPPORTED_OS_FAMILY: set[OSFamily] = {OSFamily.LINUX}
+
+    DATA_MODEL = ProcessDataModel
+    CMD_KFD = "rocm-smi --showpids"
+    CMD_CPU_USAGE = "top -b -n 1"
+    CMD_PROCESS = "top -b -n 1 -o %CPU "
+
+    def collect_data(
+        self, args: Optional[ProcessCollectorArgs] = None
+    ) -> tuple[TaskResult, Optional[ProcessDataModel]]:
+        """Collect process data from the system.
+
+        Args:
+            args (Optional[ProcessCollectorArgs], optional): process collection arguments. Defaults to None.
+
+        Returns:
+            tuple[TaskResult, Optional[ProcessDataModel]]: tuple containing the task result and the collected process data model or None if no data was collected.
+        """
+        if args is None:
+            args = ProcessCollectorArgs()
+
+        process_data = ProcessDataModel()
+        process_data.processes = []
+
+        kfd_process = self._run_sut_cmd(self.CMD_KFD)
+        if kfd_process.exit_code == 0:
+            if "No KFD PIDs currently running" in kfd_process.stdout:
+                process_data.kfd_process = 0
+            else:
+                kfd_process = re.findall(
+                    r"^\s*\d+\s+[\w]+\s+\d+\s+\d+\s+\d+\s+\d+",
+                    kfd_process.stdout,
+                    re.MULTILINE,
+                )
+                process_data.kfd_process = len(kfd_process)
+
+        cpu_usage = self._run_sut_cmd(self.CMD_CPU_USAGE)
+        if cpu_usage.exit_code == 0:
+            cpu_idle = (
+                [line for line in cpu_usage.stdout.splitlines() if "Cpu(s)" in line][0]
+                .split(",")[3]
+                .split()[0]
+                .replace("%id", "")
+            )
+            process_data.cpu_usage = 100 - float(cpu_idle)
+
+        processes = self._run_sut_cmd(
+            f"self.CMD_PROCESS | sed -n '8,{args.top_n_process + 7}p'"
+        )  # Remove system header
+        if processes.exit_code == 0:
+            for line in processes.stdout.splitlines():
+                columns = line.split()
+                process_cpu_usage = columns[8]
+                process_name = columns[11]
+                process_data.processes.append((process_name, process_cpu_usage))
+
+        process_check = bool(process_data.model_fields_set)
+        if process_check:
+            self._log_event(
+                category="PROCESS_READ",
+                description="Process data collected",
+                data=process_data.model_dump(),
+                priority=EventPriority.INFO,
+            )
+            self.result.message = "Process data collected"
+            self.result.status = ExecutionStatus.OK
+            return self.result, process_data
+        else:
+            self._log_event(
+                category=EventCategory.OS,
+                description="Process data not found",
+                priority=EventPriority.ERROR,
+            )
+            self.result.message = "Process data not found"
+            self.result.status = ExecutionStatus.ERROR
+            return self.result, None

nodescraper/plugins/inband/process/process_plugin.py

@@ -0,0 +1,46 @@
+###############################################################################
+#
+# MIT License
+#
+# Copyright (c) 2025 Advanced Micro Devices, Inc.
+#
+# Permission is hereby granted, free of charge, to any person obtaining a copy
+# of this software and associated documentation files (the "Software"), to deal
+# in the Software without restriction, including without limitation the rights
+# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+# copies of the Software, and to permit persons to whom the Software is
+# furnished to do so, subject to the following conditions:
+#
+# The above copyright notice and this permission notice shall be included in all
+# copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+# SOFTWARE.
+#
+###############################################################################
+from nodescraper.base import InBandDataPlugin
+
+from .analyzer_args import ProcessAnalyzerArgs
+from .collector_args import ProcessCollectorArgs
+from .process_analyzer import ProcessAnalyzer
+from .process_collector import ProcessCollector
+from .processdata import ProcessDataModel
+
+
+class ProcessPlugin(InBandDataPlugin[ProcessDataModel, ProcessCollectorArgs, ProcessAnalyzerArgs]):
+    """Plugin for collection and analysis of process data"""
+
+    DATA_MODEL = ProcessDataModel
+
+    COLLECTOR = ProcessCollector
+
+    ANALYZER = ProcessAnalyzer
+
+    ANALYZER_ARGS = ProcessAnalyzerArgs
+
+    COLLECTOR_ARGS = ProcessCollectorArgs

nodescraper/plugins/inband/process/processdata.py

@@ -0,0 +1,34 @@
+###############################################################################
+#
+# MIT License
+#
+# Copyright (c) 2025 Advanced Micro Devices, Inc.
+#
+# Permission is hereby granted, free of charge, to any person obtaining a copy
+# of this software and associated documentation files (the "Software"), to deal
+# in the Software without restriction, including without limitation the rights
+# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+# copies of the Software, and to permit persons to whom the Software is
+# furnished to do so, subject to the following conditions:
+#
+# The above copyright notice and this permission notice shall be included in all
+# copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+# SOFTWARE.
+#
+###############################################################################
+from typing import Optional
+
+from nodescraper.models import DataModel
+
+
+class ProcessDataModel(DataModel):
+    kfd_process: Optional[int] = None
+    cpu_usage: Optional[float] = None
+    processes: Optional[list[tuple[str, str]]] = None

nodescraper/plugins/inband/rocm/__init__.py

@@ -0,0 +1,25 @@
+###############################################################################
+#
+# MIT License
+#
+# Copyright (c) 2025 Advanced Micro Devices, Inc.
+#
+# Permission is hereby granted, free of charge, to any person obtaining a copy
+# of this software and associated documentation files (the "Software"), to deal
+# in the Software without restriction, including without limitation the rights
+# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+# copies of the Software, and to permit persons to whom the Software is
+# furnished to do so, subject to the following conditions:
+#
+# The above copyright notice and this permission notice shall be included in all
+# copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+# SOFTWARE.
+#
+###############################################################################

nodescraper/plugins/inband/rocm/analyzer_args.py

@@ -0,0 +1,66 @@
+###############################################################################
+#
+# MIT License
+#
+# Copyright (c) 2025 Advanced Micro Devices, Inc.
+#
+# Permission is hereby granted, free of charge, to any person obtaining a copy
+# of this software and associated documentation files (the "Software"), to deal
+# in the Software without restriction, including without limitation the rights
+# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+# copies of the Software, and to permit persons to whom the Software is
+# furnished to do so, subject to the following conditions:
+#
+# The above copyright notice and this permission notice shall be included in all
+# copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+# SOFTWARE.
+#
+###############################################################################
+from typing import Union
+
+from pydantic import Field, field_validator
+
+from nodescraper.models.analyzerargs import AnalyzerArgs
+from nodescraper.plugins.inband.rocm.rocmdata import RocmDataModel
+
+
+class RocmAnalyzerArgs(AnalyzerArgs):
+    exp_rocm: Union[str, list] = Field(default_factory=list)
+    exp_rocm_latest: str = Field(default="")
+
+    @field_validator("exp_rocm", mode="before")
+    @classmethod
+    def validate_exp_rocm(cls, exp_rocm: Union[str, list]) -> list:
+        """support str or list input for exp_rocm
+
+        Args:
+            exp_rocm (Union[str, list]): exp_rocm input
+
+        Returns:
+            list: exp_rocm list
+        """
+        if isinstance(exp_rocm, str):
+            exp_rocm = [exp_rocm]
+
+        return exp_rocm
+
+    @classmethod
+    def build_from_model(cls, datamodel: RocmDataModel) -> "RocmAnalyzerArgs":
+        """build analyzer args from data model
+
+        Args:
+            datamodel (RocmDataModel): data model for plugin
+
+        Returns:
+            RocmAnalyzerArgs: instance of analyzer args class
+        """
+        return cls(
+            exp_rocm=datamodel.rocm_version, exp_rocm_latest=datamodel.rocm_latest_versioned_path
+        )

nodescraper/plugins/inband/rocm/rocm_analyzer.py

@@ -0,0 +1,100 @@
+###############################################################################
+#
+# MIT License
+#
+# Copyright (c) 2025 Advanced Micro Devices, Inc.
+#
+# Permission is hereby granted, free of charge, to any person obtaining a copy
+# of this software and associated documentation files (the "Software"), to deal
+# in the Software without restriction, including without limitation the rights
+# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+# copies of the Software, and to permit persons to whom the Software is
+# furnished to do so, subject to the following conditions:
+#
+# The above copyright notice and this permission notice shall be included in all
+# copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+# SOFTWARE.
+#
+###############################################################################
+from typing import Optional
+
+from nodescraper.enums import EventCategory, EventPriority, ExecutionStatus
+from nodescraper.interfaces import DataAnalyzer
+from nodescraper.models import TaskResult
+
+from .analyzer_args import RocmAnalyzerArgs
+from .rocmdata import RocmDataModel
+
+
+class RocmAnalyzer(DataAnalyzer[RocmDataModel, RocmAnalyzerArgs]):
+    """Check ROCm matches expected versions"""
+
+    DATA_MODEL = RocmDataModel
+
+    def analyze_data(
+        self, data: RocmDataModel, args: Optional[RocmAnalyzerArgs] = None
+    ) -> TaskResult:
+        """
+        Analyze the provided ROCm data against expected versions.
+
+        Args:
+            data (RocmDataModel): ROCm data to analyze
+            args (Optional[RocmAnalyzerArgs], optional): ROCm analysis arguments. Defaults to None.
+
+        Returns:
+            TaskResult: Result of the analysis containing status and message.
+        """
+        # skip check if data not provided in config
+        if not args or not args.exp_rocm:
+            self.result.message = "Expected ROCm not provided"
+            self.result.status = ExecutionStatus.NOT_RAN
+            return self.result
+
+        for rocm_version in args.exp_rocm:
+            if data.rocm_version == rocm_version:
+                self.result.message = "ROCm version matches expected"
+                self.result.status = ExecutionStatus.OK
+                break
+        else:
+            # No matching version found
+            self.result.message = (
+                f"ROCm version mismatch! Expected: {args.exp_rocm}, actual: {data.rocm_version}"
+            )
+            self.result.status = ExecutionStatus.ERROR
+            self._log_event(
+                category=EventCategory.SW_DRIVER,
+                description=f"{self.result.message}",
+                data={"expected": args.exp_rocm, "actual": data.rocm_version},
+                priority=EventPriority.CRITICAL,
+                console_log=True,
+            )
+            return self.result
+
+        # validate rocm_latest if provided in args
+        if args.exp_rocm_latest:
+            if data.rocm_latest_versioned_path != args.exp_rocm_latest:
+                self.result.message = f"ROCm latest path mismatch! Expected: {args.exp_rocm_latest}, actual: {data.rocm_latest_versioned_path}"
+                self.result.status = ExecutionStatus.ERROR
+                self._log_event(
+                    category=EventCategory.SW_DRIVER,
+                    description=f"{self.result.message}",
+                    data={
+                        "expected": args.exp_rocm_latest,
+                        "actual": data.rocm_latest_versioned_path,
+                    },
+                    priority=EventPriority.CRITICAL,
+                    console_log=True,
+                )
+                return self.result
+            else:
+                # Update message to include rocm_latest validation result
+                self.result.message = f"ROCm version matches expected. ROCm latest path validated: {data.rocm_latest_versioned_path}"
+
+        return self.result