Skip to content

Commit f478efa

Browse files
committed
Reverse rules for Enzyme
1 parent 8bded8b commit f478efa

File tree

5 files changed

+877
-2
lines changed

5 files changed

+877
-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.109"
36+
EnzymeTestUtils = "0.2.5"
3337
JET = "0.9, 0.10"
3438
LinearAlgebra = "1"
3539
Mooncake = "0.4.183"
@@ -45,6 +49,8 @@ AMDGPU = "21141c5a-9bdb-4563-92ae-f87d6854732e"
4549
Aqua = "4c88cf16-eb10-579e-8560-4a9242c79595"
4650
ChainRulesTestUtils = "cdddcdb0-9152-4a09-a978-84456f9df70a"
4751
CUDA = "052768ef-5323-5732-b1bb-66c8b64840ba"
52+
Enzyme = "7da242da-08ed-463a-9acd-ee780be4f1d9"
53+
EnzymeTestUtils = "12d8515a-0907-448a-8884-5fe00fdf1c5a"
4854
JET = "c3a54625-cd67-489e-a8e7-0a5a0ff4e31b"
4955
Mooncake = "da2b9cff-9c12-43a0-ae48-6db2b0edb7d6"
5056
SafeTestsets = "1bc83da4-3b8d-516f-aca4-4fe02f6d838f"
@@ -54,4 +60,4 @@ TestExtras = "5ed8adda-3752-4e41-b88a-e8b09835ee3a"
5460
Zygote = "e88e6eb3-aa80-5325-afca-941959d7151f"
5561

5662
[targets]
57-
test = ["Aqua", "JET", "SafeTestsets", "Test", "TestExtras", "ChainRulesCore", "ChainRulesTestUtils", "StableRNGs", "Zygote", "CUDA", "AMDGPU", "GenericLinearAlgebra", "GenericSchur", "Mooncake"]
63+
test = ["Aqua", "JET", "SafeTestsets", "Test", "TestExtras", "ChainRulesCore", "ChainRulesTestUtils", "StableRNGs", "Zygote", "CUDA", "AMDGPU", "GenericLinearAlgebra", "GenericSchur", "Mooncake", "Enzyme", "EnzymeTestUtils"]

0 commit comments

Comments
 (0)