AFID

Author: alexandraBara

nodescraper/plugins/inband/amdsmi/amdsmi_collector.py

@@ -67,10 +67,11 @@
     StaticXgmiPlpd,
     ValueUnit,
 )
+from nodescraper.plugins.inband.amdsmi.collector_args import AmdSmiCollectorArgs
 from nodescraper.utils import get_exception_traceback
 
 
-class AmdSmiCollector(InBandDataCollector[AmdSmiDataModel, None]):
+class AmdSmiCollector(InBandDataCollector[AmdSmiDataModel, AmdSmiCollectorArgs]):
     """Class for collection of inband tool amd-smi data."""
 
     AMD_SMI_EXE = "amd-smi"
@@ -87,6 +88,7 @@ class AmdSmiCollector(InBandDataCollector[AmdSmiDataModel, None]):
     CMD_STATIC = "static -g all --json"
     CMD_STATIC_GPU = "static -g {gpu_id} --json"
     CMD_RAS = "ras --cper --folder={folder}"
+    CMD_RAS_AFID = "ras --afid --cper-file {cper_file}"
 
     def _check_amdsmi_installed(self) -> bool:
         """Check if amd-smi is installed
@@ -1266,14 +1268,57 @@ def get_cper_data(self) -> List[FileModel]:
             )
             return []
 
+    def _get_cper_afid(self, cper_file_path: str) -> Optional[int]:
+        """Get AFID from a CPER file
+
+        Args:
+            cper_file_path (str): Path to the CPER file
+
+        Returns:
+            Optional[int]: AFID value or None
+        """
+        cmd = self.CMD_RAS_AFID.format(cper_file=cper_file_path)
+        result = self._run_amd_smi(cmd)
+
+        if result is None:
+            self._log_event(
+                category=EventCategory.APPLICATION,
+                description=f"Failed to get AFID from CPER file: {cper_file_path}",
+                priority=EventPriority.ERROR,
+                console_log=True,
+            )
+            return None
+
+        try:
+            afid = int(result.strip())
+            self._log_event(
+                category=EventCategory.APPLICATION,
+                description=f"Successfully retrieved AFID from CPER file: {cper_file_path}",
+                data={"afid": afid, "cper_file": cper_file_path},
+                priority=EventPriority.INFO,
+                console_log=True,
+            )
+            return afid
+        except ValueError:
+            self._log_event(
+                category=EventCategory.APPLICATION,
+                description=f"Failed to parse AFID value from output: {result}",
+                data={"output": result, "cper_file": cper_file_path},
+                priority=EventPriority.ERROR,
+                console_log=True,
+            )
+            return None
+
     def collect_data(
         self,
-        args: Any = None,
+        args: Optional[AmdSmiCollectorArgs] = None,
     ) -> tuple[TaskResult, Optional[AmdSmiDataModel]]:
         """Collect AmdSmi data from system
 
         Args:
-            args (Any, optional): optional arguments for data collection. Defaults to None.
+            args (Optional[AmdSmiCollectorArgs], optional): optional arguments for data collection.
+                If cper_file_path is provided, will run amd-smi ras --afid --cper-file command.
+                Defaults to None.
 
         Returns:
             tuple[TaskResult, Optional[AmdSmiDataModel]]: task result and collected data model
@@ -1300,6 +1345,11 @@ def collect_data(
             if amd_smi_data is None:
                 return self.result, None
 
+            # If cper_file_path is provided, get AFID from the CPER file
+            if args and args.cper_file_path:
+                afid = self._get_cper_afid(args.cper_file_path)
+                amd_smi_data.cper_afid = afid
+
             return self.result, amd_smi_data
         except Exception as e:
             self._log_event(

nodescraper/plugins/inband/amdsmi/amdsmi_plugin.py

@@ -29,15 +29,18 @@
 from .amdsmi_collector import AmdSmiCollector
 from .amdsmidata import AmdSmiDataModel
 from .analyzer_args import AmdSmiAnalyzerArgs
+from .collector_args import AmdSmiCollectorArgs
 
 
-class AmdSmiPlugin(InBandDataPlugin[AmdSmiDataModel, None, AmdSmiAnalyzerArgs]):
+class AmdSmiPlugin(InBandDataPlugin[AmdSmiDataModel, AmdSmiCollectorArgs, AmdSmiAnalyzerArgs]):
     """Plugin for collection and analysis of amdsmi data"""
 
     DATA_MODEL = AmdSmiDataModel
 
     COLLECTOR = AmdSmiCollector
 
+    COLLECTOR_ARGS = AmdSmiCollectorArgs
+
     ANALYZER = AmdSmiAnalyzer
 
     ANALYZER_ARGS = AmdSmiAnalyzerArgs

nodescraper/plugins/inband/amdsmi/amdsmidata.py

@@ -954,6 +954,7 @@ class AmdSmiDataModel(DataModel):
     xgmi_metric: Optional[list[XgmiMetrics]] = Field(default_factory=list)
     xgmi_link: Optional[list[XgmiLinks]] = Field(default_factory=list)
     cper_data: Optional[list[FileModel]] = Field(default_factory=list)
+    cper_afid: Optional[int] = None
     amdsmitst_data: AmdSmiTstData = Field(default_factory=AmdSmiTstData)
 
     def get_list(self, gpu: int) -> Optional[AmdSmiListItem]:

nodescraper/plugins/inband/amdsmi/collector_args.py

@@ -0,0 +1,34 @@
+###############################################################################
+#
+# MIT License
+#
+# Copyright (c) 2026 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 CollectorArgs
+
+
+class AmdSmiCollectorArgs(CollectorArgs):
+    """Collector arguments for AmdSmiPlugin"""
+
+    cper_file_path: Optional[str] = None