From 6094fab8853ff38ea9e5b7aaf6eee8d1c928caef Mon Sep 17 00:00:00 2001 From: Avik Pal Date: Wed, 19 Jul 2023 20:27:11 -0400 Subject: [PATCH 1/3] Update Project.toml --- Project.toml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/Project.toml b/Project.toml index dcca344..e80e297 100644 --- a/Project.toml +++ b/Project.toml @@ -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" From 122b5610f5d0a5add9a72582d927873deeab4607 Mon Sep 17 00:00:00 2001 From: Avik Pal Date: Wed, 19 Jul 2023 20:28:02 -0400 Subject: [PATCH 2/3] Update Project.toml --- Project.toml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/Project.toml b/Project.toml index e80e297..a18f3c3 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.3" +version = "0.1.4" [deps] Adapt = "79e6a3ab-5dfb-504d-930d-738a2a938a0e" From 9211e206637914f3452ed42308be38a5f3ffec20 Mon Sep 17 00:00:00 2001 From: Avik Pal Date: Wed, 19 Jul 2023 20:33:32 -0400 Subject: [PATCH 3/3] Use `mtl` instead of private structs --- ext/LuxDeviceUtilsMetalExt.jl | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ext/LuxDeviceUtilsMetalExt.jl b/ext/LuxDeviceUtilsMetalExt.jl index abfb897..505107d 100644 --- 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