Allow user to supply rocm version via ROCM_VER env variable if not autodetected

due to missing .info directory (#56).

Signed-off-by: Karl W. Schulz <karl.schulz@amd.com>

src/utils/specs.py

@@ -186,14 +186,23 @@ def get_machine_specs(devicenum):
             break
 
     if not rocmFound:
-        _rocm_path = os.getenv("ROCM_PATH", "/opt/rocm")
-        print("Error: Unable to detect a complete local ROCm installation.")
-        print(
-            "\nThe expected %s/.info/ versioning directory is missing. Please"
-            % _rocm_path
-        )
-        print("ensure you have valid ROCm installation.")
-        sys.exit(1)
+        # check if ROCM_VER is supplied externally
+        ROCM_VER_USER = os.getenv("ROCM_VER")
+        if ROCM_VER_USER is not None:
+            print(
+                "Overriding missing ROCm version detection with ROCM_VER = %s"
+                % ROCM_VER_USER
+            )
+            rocm_ver = ROCM_VER_USER
+        else:
+            _rocm_path = os.getenv("ROCM_PATH", "/opt/rocm")
+            print("Error: Unable to detect a complete local ROCm installation.")
+            print(
+                "\nThe expected %s/.info/ versioning directory is missing. Please"
+                % _rocm_path
+            )
+            print("ensure you have valid ROCm installation.")
+            sys.exit(1)
 
     (
         gpu_id,