Skip to content

Commit cb92045

Browse files
committed
Change to enzymecore ext
1 parent 12408c0 commit cb92045

File tree

3 files changed

+4
-6
lines changed

3 files changed

+4
-6
lines changed

Project.toml

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,6 @@ version = "0.9.6"
66
Adapt = "79e6a3ab-5dfb-504d-930d-738a2a938a0e"
77
Atomix = "a9b6321e-bd34-4604-b9c9-b65b8de01458"
88
ChainRulesCore = "d360d2e6-b24c-11e9-a2a3-2a2ae2dbcce4"
9-
EnzymeCore = "f151be2c-9106-41f4-ab19-57ee4f262869"
109
GPUArraysCore = "46192b85-c4d5-4398-a991-12ede77f4527"
1110
KernelAbstractions = "63c18a36-062a-441e-b654-da1e3ab1ce7c"
1211
LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
@@ -17,15 +16,15 @@ Statistics = "10745b16-79ce-11e8-11f9-7d13ad32a3b2"
1716

1817
[weakdeps]
1918
AMDGPU = "21141c5a-9bdb-4563-92ae-f87d6854732e"
20-
Enzyme = "7da242da-08ed-463a-9acd-ee780be4f1d9"
2119
cuDNN = "02a925ec-e4fe-4b08-9a7e-0d78e3d38ccd"
2220
CUDA = "052768ef-5323-5732-b1bb-66c8b64840ba"
21+
EnzymeCore = "f151be2c-9106-41f4-ab19-57ee4f262869"
2322

2423
[extensions]
2524
NNlibAMDGPUExt = "AMDGPU"
2625
NNlibCUDAExt = "CUDA"
2726
NNlibCUDACUDNNExt = ["CUDA", "cuDNN"]
28-
NNlibEnzymeExt = "Enzyme"
27+
NNlibEnzymeCoreExt = "EnzymeCore"
2928

3029
[compat]
3130
AMDGPU = "0.5, 0.6"
@@ -34,7 +33,6 @@ Atomix = "0.1"
3433
ChainRulesCore = "1.13"
3534
CUDA = "4, 5"
3635
cuDNN = "1"
37-
Enzyme = "0.11.8"
3836
GPUArraysCore = "0.1"
3937
KernelAbstractions = "0.9.2"
4038
Requires = "1.0"

src/NNlib.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -124,8 +124,8 @@ include("impl/pooling_direct.jl")
124124
include("deprecations.jl")
125125

126126
@init @static if !isdefined(Base, :get_extension)
127-
@require Enzyme="7da242da-08ed-463a-9acd-ee780be4f1d9" begin
128-
include("../ext/NNlibEnzymeExt/NNlibEnzymeExt.jl")
127+
@require EnzymeCore="f151be2c-9106-41f4-ab19-57ee4f262869" begin
128+
include("../ext/NNlibEnzymeCoreExt/NNlibEnzymeCoresExt.jl")
129129
end
130130
end
131131

0 commit comments

Comments
 (0)