Skip to content

Commit ec8354b

Browse files
committed
Reverse rules for Enzyme
1 parent 4fbc3bf commit ec8354b

File tree

10 files changed

+946
-47
lines changed

10 files changed

+946
-47
lines changed

Project.toml

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,18 +10,22 @@ LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
1010
AMDGPU = "21141c5a-9bdb-4563-92ae-f87d6854732e"
1111
ChainRulesCore = "d360d2e6-b24c-11e9-a2a3-2a2ae2dbcce4"
1212
CUDA = "052768ef-5323-5732-b1bb-66c8b64840ba"
13+
Enzyme = "7da242da-08ed-463a-9acd-ee780be4f1d9"
1314

1415
[extensions]
1516
MatrixAlgebraKitChainRulesCoreExt = "ChainRulesCore"
1617
MatrixAlgebraKitAMDGPUExt = "AMDGPU"
1718
MatrixAlgebraKitCUDAExt = "CUDA"
19+
MatrixAlgebraKitEnzymeExt = "Enzyme"
1820

1921
[compat]
2022
AMDGPU = "2"
2123
Aqua = "0.6, 0.7, 0.8"
2224
ChainRulesCore = "1"
2325
ChainRulesTestUtils = "1"
2426
CUDA = "5"
27+
Enzyme = "0.13.77"
28+
EnzymeTestUtils = "0.2.3"
2529
JET = "0.9, 0.10"
2630
LinearAlgebra = "1"
2731
SafeTestsets = "0.1"
@@ -34,6 +38,7 @@ julia = "1.10"
3438
[extras]
3539
Aqua = "4c88cf16-eb10-579e-8560-4a9242c79595"
3640
ChainRulesTestUtils = "cdddcdb0-9152-4a09-a978-84456f9df70a"
41+
EnzymeTestUtils = "12d8515a-0907-448a-8884-5fe00fdf1c5a"
3742
JET = "c3a54625-cd67-489e-a8e7-0a5a0ff4e31b"
3843
SafeTestsets = "1bc83da4-3b8d-516f-aca4-4fe02f6d838f"
3944
StableRNGs = "860ef19b-820b-49d6-a774-d7a799459cd3"
@@ -42,4 +47,4 @@ TestExtras = "5ed8adda-3752-4e41-b88a-e8b09835ee3a"
4247
Zygote = "e88e6eb3-aa80-5325-afca-941959d7151f"
4348

4449
[targets]
45-
test = ["Aqua", "JET", "SafeTestsets", "Test", "TestExtras","ChainRulesCore", "ChainRulesTestUtils", "StableRNGs", "Zygote", "CUDA", "AMDGPU"]
50+
test = ["Aqua", "JET", "SafeTestsets", "Test", "TestExtras","ChainRulesCore", "ChainRulesTestUtils", "StableRNGs", "Zygote", "CUDA", "AMDGPU", "Enzyme", "EnzymeTestUtils"]

0 commit comments

Comments
 (0)