Fix return value of find_roc_path when ROCM_PATH is set (#798)

* Fix return value of `find_roc_path` when `ROCM_PATH` is set

* Update src/discovery/utils.jl

Co-authored-by: Simeon David Schaub <simeon@schaub.rocks>

* Bump version number

* Fix comment for `find_roc_path`

The function is really returning the libdir, not the prefix.

---------

Co-authored-by: Simeon David Schaub <simeon@schaub.rocks>

Author: giordano

Project.toml

@@ -1,6 +1,6 @@
 name = "AMDGPU"
 uuid = "21141c5a-9bdb-4563-92ae-f87d6854732e"
-version = "1.3.5"
+version = "1.3.6"
 authors = ["Julian P Samaroo <jpsamaroo@jpsamaroo.me>", "Valentin Churavy <v.churavy@gmail.com>", "Anton Smirnov <tonysmn97@gmail.com>"]
 
 [deps]

src/discovery/utils.jl

@@ -15,10 +15,13 @@ function check_rocm_path(path::String)
     return path2
 end
 
-# Find root ROCm directory.
+# Find libdir of ROCm directory.
 function find_roc_path()::String
     env_dir = get(ENV, "ROCM_PATH", "")
-    isdir(env_dir) && check_rocm_path(env_dir) != "" && return env_dir
+    if isdir(env_dir)
+    	rocm_path = check_rocm_path(env_dir)
+    	rocm_path != "" && return rocm_path
+    end
 
     if Sys.islinux()
         hipconfig = Sys.which("hipconfig")