Skip to content

Commit 7478c90

Browse files
committed
Reverse rules for Enzyme
1 parent 58846bb commit 7478c90

File tree

10 files changed

+1003
-50
lines changed

10 files changed

+1003
-50
lines changed

Project.toml

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,13 +12,15 @@ ChainRulesCore = "d360d2e6-b24c-11e9-a2a3-2a2ae2dbcce4"
1212
CUDA = "052768ef-5323-5732-b1bb-66c8b64840ba"
1313
GenericLinearAlgebra = "14197337-ba66-59df-a3e3-ca00e7dcff7a"
1414
GenericSchur = "c145ed77-6b09-5dd9-b285-bf645a82121e"
15+
Enzyme = "7da242da-08ed-463a-9acd-ee780be4f1d9"
1516

1617
[extensions]
1718
MatrixAlgebraKitChainRulesCoreExt = "ChainRulesCore"
1819
MatrixAlgebraKitAMDGPUExt = "AMDGPU"
1920
MatrixAlgebraKitCUDAExt = "CUDA"
2021
MatrixAlgebraKitGenericLinearAlgebraExt = "GenericLinearAlgebra"
2122
MatrixAlgebraKitGenericSchurExt = "GenericSchur"
23+
MatrixAlgebraKitEnzymeExt = "Enzyme"
2224

2325
[compat]
2426
AMDGPU = "2"
@@ -28,6 +30,8 @@ ChainRulesTestUtils = "1"
2830
CUDA = "5"
2931
GenericLinearAlgebra = "0.3.19"
3032
GenericSchur = "0.5.6"
33+
Enzyme = "0.13.102"
34+
EnzymeTestUtils = "0.2.5"
3135
JET = "0.9, 0.10"
3236
LinearAlgebra = "1"
3337
SafeTestsets = "0.1"
@@ -42,6 +46,7 @@ AMDGPU = "21141c5a-9bdb-4563-92ae-f87d6854732e"
4246
Aqua = "4c88cf16-eb10-579e-8560-4a9242c79595"
4347
ChainRulesTestUtils = "cdddcdb0-9152-4a09-a978-84456f9df70a"
4448
CUDA = "052768ef-5323-5732-b1bb-66c8b64840ba"
49+
EnzymeTestUtils = "12d8515a-0907-448a-8884-5fe00fdf1c5a"
4550
JET = "c3a54625-cd67-489e-a8e7-0a5a0ff4e31b"
4651
SafeTestsets = "1bc83da4-3b8d-516f-aca4-4fe02f6d838f"
4752
StableRNGs = "860ef19b-820b-49d6-a774-d7a799459cd3"
@@ -50,4 +55,4 @@ TestExtras = "5ed8adda-3752-4e41-b88a-e8b09835ee3a"
5055
Zygote = "e88e6eb3-aa80-5325-afca-941959d7151f"
5156

5257
[targets]
53-
test = ["Aqua", "JET", "SafeTestsets", "Test", "TestExtras", "ChainRulesCore", "ChainRulesTestUtils", "StableRNGs", "Zygote", "CUDA", "AMDGPU", "GenericLinearAlgebra", "GenericSchur"]
58+
test = ["Aqua", "JET", "SafeTestsets", "Test", "TestExtras", "ChainRulesCore", "ChainRulesTestUtils", "StableRNGs", "Zygote", "CUDA", "AMDGPU", "GenericLinearAlgebra", "GenericSchur", "Enzyme", "EnzymeTestUtils"]

0 commit comments

Comments
 (0)