Merge pull request #8 from LuxDL/ap/metal

Fixes #7

Author: avik-pal

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"

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