Print amdgpu version

Change-Id: Ibab6e51489d436b66c3bac4bbd0f52a400ad6b0b
ROCm · Oct 3, 2023 · b249107 · b249107
1 parent f1f463d
commit b249107
Showing 1 changed file with 11 additions and 1 deletion.
diff --git 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);