From b249107c6ab443f2ec9131215f39a4fde4e8e240 Mon Sep 17 00:00:00 2001
From: Gregory Rodgers <Gregory.Rodgers@amd.com>
Date: Tue, 3 Oct 2023 18:47:21 +0000
Subject: [PATCH] Print amdgpu version

Change-Id: Ibab6e51489d436b66c3bac4bbd0f52a400ad6b0b
---
 rocminfo.cc | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/rocminfo.cc b/rocminfo.cc
index f3c8e89d64..4b7b287978 100755
--- a/rocminfo.cc
+++ b/rocminfo.cc
@@ -1118,7 +1118,17 @@ int CheckInitialState(void) {
       }
     }
     if (is_live){
-      printf("%sROCk module is loaded%s\n", COL_WHT, COL_RESET);
+      std::ifstream amdgpu_version("/sys/module/amdgpu/version");
+      if (amdgpu_version){
+        std::stringstream buffer;
+        buffer << amdgpu_version.rdbuf();
+        std::string vers;
+        std::getline(buffer, vers);
+        amdgpu_version.close();
+        printf("%sROCk module version %s is loaded%s\n", COL_WHT, vers.c_str(), COL_RESET);
+      } else {
+        printf("%sROCk module is loaded%s\n", COL_WHT, COL_RESET);
+      }
     } else {
       printf("%sROCk module is NOT live, possibly no GPU devices%s\n",
                                                           COL_RED, COL_RESET);