Skip to content
This repository has been archived by the owner on Nov 4, 2024. It is now read-only.

More verbose warning if trigger package not loaded #17

Merged
merged 1 commit into from
Oct 20, 2023
Merged

More verbose warning if trigger package not loaded #17

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion Project.toml
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
name = "LuxDeviceUtils"
uuid = "34f89e08-e1d5-43b4-8944-0b49ac560553"
authors = ["Avik Pal <[email protected]> and contributors"]
version = "0.1.8"
version = "0.1.9"

[deps]
Adapt = "79e6a3ab-5dfb-504d-930d-738a2a938a0e"
Expand Down
5 changes: 4 additions & 1 deletion src/LuxDeviceUtils.jl
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -158,7 +158,10 @@ function _get_gpu_device(; force_gpu_usage::Bool)
@warn """No functional GPU backend found! Defaulting to CPU.

1. If no GPU is available, nothing needs to be done.
2. If GPU is available, load the corresponding trigger package.""" maxlog=1
2. If GPU is available, load the corresponding trigger package.
a. LuxCUDA.jl for NVIDIA CUDA Support!
b. LuxAMDGPU.jl for AMD GPU ROCM Support!
c. Metal.jl for Apple Metal GPU Support!""" maxlog=1
return cpu_device()
end
end
Expand Down