From 75211a6389aacf2cb7757aa8dfb318470eff98d9 Mon Sep 17 00:00:00 2001
From: "Pryor, Adam" <Adam.Pryor@amd.com>
Date: Fri, 21 Feb 2025 15:17:19 -0600
Subject: [PATCH] [SWDEV-500382] Add check to pcie throughput for uint64 max
 (#120)

Signed-off-by: Adam Pryor <Adam.pryor@amd.com>
---
 rocm_smi/src/rocm_smi.cc | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/rocm_smi/src/rocm_smi.cc b/rocm_smi/src/rocm_smi.cc
index 6dcc180fbc..83c28cd160 100644
--- a/rocm_smi/src/rocm_smi.cc
+++ b/rocm_smi/src/rocm_smi.cc
@@ -3074,6 +3074,10 @@ rsmi_dev_pci_throughput_get(uint32_t dv_ind, uint64_t *sent,
     fs_rng >> *max_pkt_sz;
   }
 
+  if ((sent && *sent == UINT64_MAX) || (received && *received == UINT64_MAX)){
+    return RSMI_STATUS_NOT_SUPPORTED;
+  }
+
   return RSMI_STATUS_SUCCESS;
   CATCH
 }