revert to using separate artifact for device libs

Author: simeonschaub

Project.toml

@@ -26,6 +26,7 @@ Pkg = "44cfe95a-1eb2-52ea-b672-e2afdf69b78f"
 Preferences = "21216c6a-2e73-6563-6e65-726566657250"
 PrettyTables = "08abe8d2-0d0c-5749-adfa-8a2ac140af0d"
 Printf = "de0858da-6303-5e67-8744-51eddeeeb8d7"
+ROCmDeviceLibs_jll = "873c0968-716b-5aa7-bb8d-d1e2e2aeff2d"
 Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"
 Random123 = "74087812-796a-5b5d-8853-05524746bad3"
 RandomNumbers = "e6cf234a-135c-5ec9-84dd-332b85af5143"
@@ -65,6 +66,7 @@ LazyArtifacts = "1.10"
 Libdl = "1.10"
 Preferences = "1"
 PrettyTables = "3"
+ROCmDeviceLibs_jll = "=5.6.1, =6.2.1, =7.0.2"
 Random123 = "1.6"
 RandomNumbers = "1.5"
 SpecialFunctions = "2"

src/discovery/discovery.jl

@@ -5,6 +5,7 @@ export librocblas, librocsparse, librocsolver
 export librocrand, librocfft, libMIOpen_path
 
 using LLD_jll
+using ROCmDeviceLibs_jll
 using LazyArtifacts
 using Preferences
 using Libdl
@@ -26,6 +27,14 @@ function get_ld_lld(rocm_path::String)::Tuple{String, Bool}
     return (LLD_jll.lld_path, true)
 end
 
+function get_device_libs(from_artifact::Bool; rocm_path::String)
+    if from_artifact && ROCmDeviceLibs_jll.is_available()
+        ROCmDeviceLibs_jll.bitcode_path
+    else
+        find_device_libs(rocm_path)
+    end
+end
+
 function _hip_runtime_version()
     v_ref = Ref{Cint}()
     res = ccall((:hipRuntimeGetVersion, libhip), UInt32, (Ptr{Cint},), v_ref)
@@ -80,7 +89,9 @@ function __init__()
         global lld_artifact = lld_artifact
         global libhip = find_rocm_library(Sys.islinux() ? "libamdhip64" : "amdhip64"; rocm_path)
 
-        global libdevice_libs = find_device_libs(rocm_path)
+        # Always load artifact device libraries.
+        from_artifact = true
+        global libdevice_libs = get_device_libs(from_artifact; rocm_path)
 
         # HIP-based libraries.
         global librocblas = find_rocm_library(lib_prefix * "rocblas"; rocm_path)