Skip to content

Commit c78e81e

Browse files
committed
Reverse rules for Enzyme
1 parent 91a8a69 commit c78e81e

File tree

6 files changed

+903
-2
lines changed

6 files changed

+903
-2
lines changed

Project.toml

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,7 @@ 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
GenericLinearAlgebra = "14197337-ba66-59df-a3e3-ca00e7dcff7a"
1415
GenericSchur = "c145ed77-6b09-5dd9-b285-bf645a82121e"
1516
Mooncake = "da2b9cff-9c12-43a0-ae48-6db2b0edb7d6"
@@ -18,6 +19,7 @@ Mooncake = "da2b9cff-9c12-43a0-ae48-6db2b0edb7d6"
1819
MatrixAlgebraKitChainRulesCoreExt = "ChainRulesCore"
1920
MatrixAlgebraKitAMDGPUExt = "AMDGPU"
2021
MatrixAlgebraKitCUDAExt = "CUDA"
22+
MatrixAlgebraKitEnzymeExt = "Enzyme"
2123
MatrixAlgebraKitGenericLinearAlgebraExt = "GenericLinearAlgebra"
2224
MatrixAlgebraKitGenericSchurExt = "GenericSchur"
2325
MatrixAlgebraKitMooncakeExt = "Mooncake"
@@ -30,6 +32,8 @@ ChainRulesTestUtils = "1"
3032
CUDA = "5"
3133
GenericLinearAlgebra = "0.3.19"
3234
GenericSchur = "0.5.6"
35+
Enzyme = "0.13.112"
36+
EnzymeTestUtils = "0.2.5"
3337
JET = "0.9, 0.10"
3438
LinearAlgebra = "1"
3539
Mooncake = "0.4.183"
@@ -46,6 +50,8 @@ AMDGPU = "21141c5a-9bdb-4563-92ae-f87d6854732e"
4650
Aqua = "4c88cf16-eb10-579e-8560-4a9242c79595"
4751
ChainRulesTestUtils = "cdddcdb0-9152-4a09-a978-84456f9df70a"
4852
CUDA = "052768ef-5323-5732-b1bb-66c8b64840ba"
53+
Enzyme = "7da242da-08ed-463a-9acd-ee780be4f1d9"
54+
EnzymeTestUtils = "12d8515a-0907-448a-8884-5fe00fdf1c5a"
4955
JET = "c3a54625-cd67-489e-a8e7-0a5a0ff4e31b"
5056
Mooncake = "da2b9cff-9c12-43a0-ae48-6db2b0edb7d6"
5157
Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"
@@ -57,4 +63,4 @@ Zygote = "e88e6eb3-aa80-5325-afca-941959d7151f"
5763

5864
[targets]
5965
test = ["Aqua", "JET", "SafeTestsets", "Test", "TestExtras", "ChainRulesCore",
60-
"ChainRulesTestUtils", "Random", "StableRNGs", "Zygote", "CUDA", "AMDGPU", "GenericLinearAlgebra", "GenericSchur", "Mooncake"]
66+
"ChainRulesTestUtils", "Random", "StableRNGs", "Zygote", "CUDA", "AMDGPU", "GenericLinearAlgebra", "GenericSchur", "Mooncake", "Enzyme", "EnzymeTestUtils"]

0 commit comments

Comments
 (0)