diff --git a/Project.toml b/Project.toml index cb37f72..6c82698 100644 --- a/Project.toml +++ b/Project.toml @@ -1,7 +1,7 @@ name = "LuxDeviceUtils" uuid = "34f89e08-e1d5-43b4-8944-0b49ac560553" authors = ["Avik Pal and contributors"] -version = "0.1.8" +version = "0.1.9" [deps] Adapt = "79e6a3ab-5dfb-504d-930d-738a2a938a0e" diff --git a/src/LuxDeviceUtils.jl b/src/LuxDeviceUtils.jl index a4ff46f..a45379a 100644 --- a/src/LuxDeviceUtils.jl +++ b/src/LuxDeviceUtils.jl @@ -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