Skip to content

Commit 2e2ba4c

Browse files
authored
Merge pull request #17 from LuxDL/ap/warn
More verbose warning if trigger package not loaded
2 parents c2ee4a8 + 061b194 commit 2e2ba4c

File tree

2 files changed

+5
-2
lines changed

2 files changed

+5
-2
lines changed

lib/MLDataDevices/Project.toml

+1-1
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
name = "LuxDeviceUtils"
22
uuid = "34f89e08-e1d5-43b4-8944-0b49ac560553"
33
authors = ["Avik Pal <avikpal@mit.edu> and contributors"]
4-
version = "0.1.8"
4+
version = "0.1.9"
55

66
[deps]
77
Adapt = "79e6a3ab-5dfb-504d-930d-738a2a938a0e"

lib/MLDataDevices/src/LuxDeviceUtils.jl

+4-1
Original file line numberDiff line numberDiff line change
@@ -158,7 +158,10 @@ function _get_gpu_device(; force_gpu_usage::Bool)
158158
@warn """No functional GPU backend found! Defaulting to CPU.
159159
160160
1. If no GPU is available, nothing needs to be done.
161-
2. If GPU is available, load the corresponding trigger package.""" maxlog=1
161+
2. If GPU is available, load the corresponding trigger package.
162+
a. LuxCUDA.jl for NVIDIA CUDA Support!
163+
b. LuxAMDGPU.jl for AMD GPU ROCM Support!
164+
c. Metal.jl for Apple Metal GPU Support!""" maxlog=1
162165
return cpu_device()
163166
end
164167
end

0 commit comments

Comments
 (0)