Merge pull request #8 from LuxDL/ap/metal

Fixes #7
LuxDL · Jul 20, 2023 · 10ccab1 · 10ccab1 · avik-pal · Jul 20, 2023
2 parents 221e3c5 + 9211e20
commit 10ccab1
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/Project.toml b/Project.toml
@@ -1,7 +1,7 @@
 name = "LuxDeviceUtils"
 uuid = "34f89e08-e1d5-43b4-8944-0b49ac560553"
 authors = ["Avik Pal <[email protected]> and contributors"]
-version = "0.1.3"
+version = "0.1.4"
 
 [deps]
 Adapt = "79e6a3ab-5dfb-504d-930d-738a2a938a0e"
@@ -36,7 +36,7 @@ Functors = "0.2, 0.3, 0.4"
 LuxAMDGPU = "0.1"
 LuxCUDA = "0.2, 0.3"
 LuxCore = "0.1.4"
-Metal = "0.4"
+Metal = "0.4, 0.5"
 PackageExtensionCompat = "1"
 Preferences = "1"
 TruncatedStacktraces = "1"

diff --git a/ext/LuxDeviceUtilsMetalExt.jl b/ext/LuxDeviceUtilsMetalExt.jl
@@ -8,7 +8,7 @@ __init__() = reset_gpu_device!()
 
 # Device Transfer
 ## To GPU
-adapt_storage(::LuxMetalAdaptor, x) = adapt_storage(Metal.MtlArrayAdaptor(), x)
+adapt_storage(::LuxMetalAdaptor, x) = mtl(x)
 adapt_storage(::LuxMetalAdaptor, rng::AbstractRNG) = rng
 
 ## Chain Rules