diff --git a/projects/amdsmi/amdsmi_cli/amdsmi_commands.py b/projects/amdsmi/amdsmi_cli/amdsmi_commands.py index 15a36bb351..737c87d215 100644 --- a/projects/amdsmi/amdsmi_cli/amdsmi_commands.py +++ b/projects/amdsmi/amdsmi_cli/amdsmi_commands.py @@ -6381,6 +6381,20 @@ class AMDSMICommands(): if args.follow and not getattr(self, "_cper_follow_prompted", False): print("Press CTRL + C to stop.") self._cper_follow_prompted = True + + partition_id = -1 + try: + kfd_info = amdsmi_interface.amdsmi_get_gpu_kfd_info(args.gpu) + kfd_id = kfd_info['kfd_id'] + node_id = kfd_info['node_id'] + partition_id = kfd_info['current_partition_id'] + except amdsmi_exception.AmdSmiLibraryException as e: + logging.debug("Failed to get kfd info for gpu %s | %s", gpu_id, e.get_error_info()) + + if partition_id != 0: + logging.debug(f"Skipping gpu {gpu_id} on non zero partition {partition_id}") + return + if args.folder and args.gpu: print(f"Dumping CPER file header entries for GPU {gpu_id} in folder {args.folder}") elif args.folder: