From bd55b27db2505379069aef2f606c256388586ca5 Mon Sep 17 00:00:00 2001 From: Avik Pal Date: Fri, 20 Oct 2023 12:26:38 -0400 Subject: [PATCH] A more verbose warning --- Project.toml | 2 +- src/LuxDeviceUtils.jl | 5 ++++- 2 files changed, 5 insertions(+), 2 deletions(-) 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