Skip to content

Commit bb6d623

Browse files
wsmosesChrisRackauckas
authored andcommitted
Add Enzyme extension
1 parent 5a25b7d commit bb6d623

File tree

2 files changed

+6
-0
lines changed

2 files changed

+6
-0
lines changed

Project.toml

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,7 @@ ArrayInterface = "4fba245c-0d91-5ea0-9b3e-6abc04ee57a9"
88
ConcreteStructs = "2569d6c7-a4a2-43d3-a901-331e8e4be471"
99
DocStringExtensions = "ffbed154-4ef7-542d-bbb7-c09d3a79fcae"
1010
EnumX = "4e289a0a-7415-4d19-859d-a7e5c4648b56"
11+
EnzymeCore = "f151be2c-9106-41f4-ab19-57ee4f262869"
1112
FastLapackInterface = "29a986be-02c6-4525-aec4-84b980013641"
1213
GPUArraysCore = "46192b85-c4d5-4398-a991-12ede77f4527"
1314
InteractiveUtils = "b77e0a4c-d291-57a0-90e8-8db25a27a240"
@@ -30,6 +31,7 @@ UnPack = "3a884ed6-31ef-47d7-9d2a-63182c4928ed"
3031

3132
[weakdeps]
3233
BlockDiagonals = "0a1fb500-61f7-11e9-3c65-f5ef3456f9f0"
34+
Enzyme = "7da242da-08ed-463a-9acd-ee780be4f1d9"
3335
CUDA = "052768ef-5323-5732-b1bb-66c8b64840ba"
3436
HYPRE = "b5ffcf37-a2bd-41ab-a3da-4bd9bc8ad771"
3537
IterativeSolvers = "42fd0dbc-a981-5370-80f2-aaf504508153"
@@ -42,6 +44,7 @@ Pardiso = "46dd5b70-b6fb-5a00-ae2d-e8fea33afaf2"
4244
[extensions]
4345
LinearSolveBlockDiagonalsExt = "BlockDiagonals"
4446
LinearSolveCUDAExt = "CUDA"
47+
LinearSolveEnzymeExt = "Enzyme"
4548
LinearSolveHYPREExt = "HYPRE"
4649
LinearSolveIterativeSolversExt = "IterativeSolvers"
4750
LinearSolveKernelAbstractionsExt = "KernelAbstractions"

src/init.jl

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,5 +15,8 @@ function __init__()
1515
@require MKL_jll="856f044c-d86e-5d09-b602-aeab76dc8ba7" begin
1616
include("../ext/LinearSolveMKLExt.jl")
1717
end
18+
@require Enzyme="7da242da-08ed-463a-9acd-ee780be4f1d9" begin
19+
include("../ext/LinearSolveEnzymeExt.jl")
20+
end
1821
end
1922
end

0 commit comments

Comments
 (0)