Skip to content

Commit e39cd36

Browse files
avik-palavik-pal
committed
test: exported functions
1 parent c86486d commit e39cd36

File tree

8 files changed

+224
-167
lines changed

8 files changed

+224
-167
lines changed

‎CondaPkg.toml‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,6 @@
22
python = "<=3.13,>=3.9,<4"
33

44
[pip.deps]
5-
jax = ">= 0.6"
5+
jax = ">= 0.5"
66
tensorflow = ">= 2.17"
7-
numpy = ">= 2"
7+
numpy = ">= 1, >= 2"

‎Project.toml‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -98,6 +98,7 @@ Libdl = "1.10"
9898
LinearAlgebra = "1.10"
9999
MPI = "0.20"
100100
NNlib = "0.9.26"
101+
NPZ = "0.4"
101102
OffsetArrays = "1"
102103
OneHotArrays = "0.2.10"
103104
OrderedCollections = "1"

‎test/CondaPkg.toml‎

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,3 @@
1+
[pip.deps]
2+
jax = "== 0.5"
3+
enzyme_ad = ""

‎test/Project.toml‎

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,7 @@ LuxLib = "82251201-b29d-42c6-8e01-566dec8acb11"
2121
MethodAnalysis = "85b6ec6f-f7df-4429-9514-a64bcd9ee824"
2222
MPI = "da04e1cc-30fd-572f-bb4f-1f8673147195"
2323
NNlib = "872c559c-99b0-510c-b3b7-b6c96a88d5cd"
24+
NPZ = "15e1cf62-19b3-5cfa-8e77-841668bca605"
2425
OffsetArrays = "6fe1bfb0-de20-5000-8ca7-80f57d26f881"
2526
OneHotArrays = "0b1bfda6-eb8a-41d2-88d8-f5af5cad476f"
2627
Optimisers = "3bd65402-5787-11e9-1adc-39752487f4e2"
@@ -56,6 +57,7 @@ Lux = "1.21"
5657
LuxLib = "1.11"
5758
MPI = "0.20"
5859
NNlib = "0.9.26"
60+
NPZ = "0.4"
5961
OffsetArrays = "1"
6062
OneHotArrays = "0.2.6"
6163
Optimisers = "0.4"

‎test/export_enzymeax.jl‎

Lines changed: 0 additions & 55 deletions
This file was deleted.

‎test/export_enzymeax_comprehensive.jl‎

Lines changed: 0 additions & 109 deletions
This file was deleted.

0 commit comments

Comments
 (0)