Skip to content

Commit

Permalink
test: try separating the test Project files
Browse files Browse the repository at this point in the history 
[skip docs]
  • Loading branch information
@avik-pal
avik-pal committed Aug 4, 2024
1 parent 523cf7e commit a0095bb
Show file tree
Hide file tree
Showing 5 changed files with 81 additions and 51 deletions.
2 changes: 1 addition & 1 deletion .github/workflows/CompatHelper.yml
View file
Original file line number Diff line number Diff line change
Expand Up @@ -34,7 +34,7 @@ jobs:
- name: "Run CompatHelper"
run: |
import CompatHelper
subdirs = ["", "docs"]
subdirs = ["", "docs", "test"]
append!(subdirs, joinpath.(("examples",), filter(p -> isdir(joinpath("examples", p)), readdir("examples"))))
CompatHelper.main(; subdirs)
shell: julia --color=yes {0}
Expand Down
46 changes: 2 additions & 44 deletions Project.toml
View file
Original file line number Diff line number Diff line change
Expand Up @@ -66,87 +66,45 @@ LuxZygoteExt = "Zygote"
[compat]
ADTypes = "1.5"
Adapt = "4"
Aqua = "0.8.4"
ArgCheck = "2.1"
ArrayInterface = "7.9"
Bumper = "0.6"
CUDA = "5.3.2"
ChainRules = "1.68"
ChainRulesCore = "1.24"
Compat = "4.15"
ComponentArrays = "0.15.16"
ConcreteStructs = "0.2.3"
ConstructionBase = "1.5"
DispatchDoctor = "0.4.12"
Documenter = "1.4"
DynamicExpressions = "0.16, 0.17, 0.18, 0.19"
Enzyme = "0.12.26"
EnzymeCore = "0.7.7"
ExplicitImports = "1.9.0"
FastClosures = "0.3.2"
Flux = "0.14.11"
Flux = "0.14.17"
ForwardDiff = "0.10.36"
FunctionWrappers = "1.1.3"
Functors = "0.4.11"
Functors = "0.4.12"
GPUArraysCore = "0.1.6"
Hwloc = "3.2.0"
InteractiveUtils = "<0.0.1, 1"
LinearAlgebra = "1.10"
Logging = "1.10"
LossFunctions = "0.11.1"
LuxCore = "0.1.16"
LuxDeviceUtils = "0.1.26"
LuxLib = "0.3.38"
LuxTestUtils = "1.1.2"
MLUtils = "0.4.3"
MPI = "0.20.19"
MacroTools = "0.5.13"
Markdown = "1.10"
NCCL = "0.1.1"
NNlib = "0.9.21"
OneHotArrays = "0.2.5"
Optimisers = "0.3"
Pkg = "1.10"
Preferences = "1.4.3"
Random = "1.10"
ReTestItems = "1.23.1"
Reexport = "1.2.2"
ReverseDiff = "1.15"
Setfield = "1.1.1"
SimpleChains = "0.4.7"
StableRNGs = "1"
Statistics = "1.10"
Test = "1.10"
Tracker = "0.2.34"
UnrolledUtilities = "0.1.2"
WeightInitializers = "0.1.5, 1"
Zygote = "0.6.70"
julia = "1.10"

[extras]
Aqua = "4c88cf16-eb10-579e-8560-4a9242c79595"
Bumper = "8ce10254-0962-460f-a3d8-1f77fea1446e"
ChainRules = "082447d4-558c-5d27-93f4-14fc19e9eca2"
ComponentArrays = "b0b7db55-cfe3-40fc-9ded-d10e2dbeff66"
Documenter = "e30172f5-a6a5-5a46-863b-614d45cd2de4"
DynamicExpressions = "a40a106e-89c9-4ca8-8020-a735e8728b6b"
Enzyme = "7da242da-08ed-463a-9acd-ee780be4f1d9"
ExplicitImports = "7d51a73a-1435-4ff3-83d9-f097790105c7"
Hwloc = "0e44f5e4-bd66-52a0-8798-143a42290a1d"
InteractiveUtils = "b77e0a4c-d291-57a0-90e8-8db25a27a240"
Logging = "56ddb016-857b-54e1-b83d-db4d58db5568"
LuxTestUtils = "ac9de150-d08f-4546-94fb-7472b5760531"
MLUtils = "f1d291b0-491e-4a28-83b9-f70985020b54"
OneHotArrays = "0b1bfda6-eb8a-41d2-88d8-f5af5cad476f"
Pkg = "44cfe95a-1eb2-52ea-b672-e2afdf69b78f"
ReTestItems = "817f1d60-ba6b-4fd5-9520-3cf149f6a823"
ReverseDiff = "37e2e3b7-166d-5795-8a7a-e32c996b4267"
SimpleChains = "de6bee2f-e2f4-4ec7-b6ed-219cc6f6e9e5"
StableRNGs = "860ef19b-820b-49d6-a774-d7a799459cd3"
Statistics = "10745b16-79ce-11e8-11f9-7d13ad32a3b2"
Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
Tracker = "9f7883ad-71c0-57eb-9f7f-b5c9e6d3789c"
Zygote = "e88e6eb3-aa80-5325-afca-941959d7151f"

[targets]
test = ["Aqua", "Bumper", "ComponentArrays", "Documenter", "DynamicExpressions", "Enzyme", "ExplicitImports", "Hwloc", "InteractiveUtils", "Logging", "LuxTestUtils", "MLUtils", "OneHotArrays", "Pkg", "ReTestItems", "ReverseDiff", "SimpleChains", "StableRNGs", "Statistics", "Test", "Tracker", "Zygote"]
77 changes: 77 additions & 0 deletions test/Project.toml
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,77 @@
[deps]
ADTypes = "47edcb42-4c32-4615-8424-f2b9edc5f35b"
Adapt = "79e6a3ab-5dfb-504d-930d-738a2a938a0e"
Aqua = "4c88cf16-eb10-579e-8560-4a9242c79595"
Bumper = "8ce10254-0962-460f-a3d8-1f77fea1446e"
ChainRulesCore = "d360d2e6-b24c-11e9-a2a3-2a2ae2dbcce4"
ComponentArrays = "b0b7db55-cfe3-40fc-9ded-d10e2dbeff66"
Documenter = "e30172f5-a6a5-5a46-863b-614d45cd2de4"
DynamicExpressions = "a40a106e-89c9-4ca8-8020-a735e8728b6b"
Enzyme = "7da242da-08ed-463a-9acd-ee780be4f1d9"
ExplicitImports = "7d51a73a-1435-4ff3-83d9-f097790105c7"
Flux = "587475ba-b771-5e3f-ad9e-33799f191a9c"
ForwardDiff = "f6369f11-7733-5829-9624-2563aa707210"
Functors = "d9f16b24-f501-4c13-a1f2-28368ffc5196"
Hwloc = "0e44f5e4-bd66-52a0-8798-143a42290a1d"
InteractiveUtils = "b77e0a4c-d291-57a0-90e8-8db25a27a240"
LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
Logging = "56ddb016-857b-54e1-b83d-db4d58db5568"
LuxCore = "bb33d45b-7691-41d6-9220-0943567d0623"
LuxDeviceUtils = "34f89e08-e1d5-43b4-8944-0b49ac560553"
LuxLib = "82251201-b29d-42c6-8e01-566dec8acb11"
LuxTestUtils = "ac9de150-d08f-4546-94fb-7472b5760531"
MLUtils = "f1d291b0-491e-4a28-83b9-f70985020b54"
OneHotArrays = "0b1bfda6-eb8a-41d2-88d8-f5af5cad476f"
Optimisers = "3bd65402-5787-11e9-1adc-39752487f4e2"
Pkg = "44cfe95a-1eb2-52ea-b672-e2afdf69b78f"
Preferences = "21216c6a-2e73-6563-6e65-726566657250"
Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"
ReTestItems = "817f1d60-ba6b-4fd5-9520-3cf149f6a823"
Reexport = "189a3867-3050-52da-a836-e630ba90ab69"
ReverseDiff = "37e2e3b7-166d-5795-8a7a-e32c996b4267"
Setfield = "efcf1570-3423-57d1-acb7-fd33fddbac46"
SimpleChains = "de6bee2f-e2f4-4ec7-b6ed-219cc6f6e9e5"
StableRNGs = "860ef19b-820b-49d6-a774-d7a799459cd3"
Statistics = "10745b16-79ce-11e8-11f9-7d13ad32a3b2"
Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
Tracker = "9f7883ad-71c0-57eb-9f7f-b5c9e6d3789c"
Zygote = "e88e6eb3-aa80-5325-afca-941959d7151f"

[compat]
ADTypes = "1.5"
Adapt = "4"
Aqua = "0.8.4"
Bumper = "0.6, 0.7"
ChainRulesCore = "1.24"
ComponentArrays = "0.15.16"
Documenter = "1.4"
DynamicExpressions = "0.16, 0.17, 0.18, 0.19"
Enzyme = "0.12.26"
ExplicitImports = "1.9.0"
ForwardDiff = "0.10.36"
Functors = "0.4.12"
Flux = "0.14.17"
Hwloc = "3.2.0"
InteractiveUtils = "<0.0.1, 1"
LinearAlgebra = "1.10"
Logging = "1.10"
LuxCore = "0.1.16"
LuxDeviceUtils = "0.1.26"
LuxLib = "0.3.38"
LuxTestUtils = "1.1.2"
MLUtils = "0.4.3"
OneHotArrays = "0.2.5"
Optimisers = "0.3"
Pkg = "1.10"
Preferences = "1.4.3"
Random = "1.10"
ReTestItems = "1.24.0"
Reexport = "1.2.2"
ReverseDiff = "1.15.3"
Setfield = "1.1.1"
SimpleChains = "0.4.7"
StableRNGs = "1.0.2"
Statistics = "1.11.1"
Test = "1.10"
Tracker = "0.2.34"
Zygote = "0.6.70"
2 changes: 1 addition & 1 deletion test/runtests.jl
View file
Original file line number Diff line number Diff line change
Expand Up @@ -76,7 +76,7 @@ const RETESTITEMS_NWORKERS = parse(
(tag == "distributed" || tag == "eltype_match") && continue
@info "Running tests for group: [$(i)/$(length(LUX_TEST_GROUP))] $tag"

ReTestItems.runtests(@__DIR__; tags=(tag == "all" ? nothing : [Symbol(tag)]),
ReTestItems.runtests(Lux; tags=(tag == "all" ? nothing : [Symbol(tag)]),
nworkers=RETESTITEMS_NWORKERS, testitem_timeout=3600)
end
end
Expand Down
5 changes: 0 additions & 5 deletions test/shared_testsetup.jl
View file
Original file line number Diff line number Diff line change
Expand Up @@ -5,11 +5,6 @@ include("setup_modes.jl")
import Reexport: @reexport

using Lux, Functors

# FIXME: remove after https://github.com/FluxML/Functors.jl/pull/87 is merged
(::Functors.FlattenWalk)(f::F, ::NamedTuple{()}) where {F} = []
(::Functors.FlattenWalk)(f::F, ::@NamedTuple{}) where {F} = []

@reexport using ComponentArrays, LuxCore, LuxLib, LuxTestUtils, Random, StableRNGs, Test,
Zygote, Statistics, Enzyme, LinearAlgebra, ForwardDiff
using LuxTestUtils: check_approx
Expand Down

1 comment on commit a0095bb

@github-actions
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Benchmark Results

Benchmark suite Current: a0095bb Previous: 22b87b5 Ratio
Dense(2 => 2)/cpu/reverse/ReverseDiff (compiled)/(2, 128) 4210.375 ns 3678.125 ns 1.14
Dense(2 => 2)/cpu/reverse/Zygote/(2, 128) 7280.857142857143 ns 6732.571428571428 ns 1.08
Dense(2 => 2)/cpu/reverse/Tracker/(2, 128) 21329.5 ns 21270 ns 1.00
Dense(2 => 2)/cpu/reverse/ReverseDiff/(2, 128) 9790.4 ns 9706 ns 1.01
Dense(2 => 2)/cpu/reverse/Flux/(2, 128) 9162.25 ns 9059 ns 1.01
Dense(2 => 2)/cpu/reverse/SimpleChains/(2, 128) 4474.625 ns 4460.875 ns 1.00
Dense(2 => 2)/cpu/reverse/Enzyme/(2, 128) 4985.5 ns 4948.6875 ns 1.01
Dense(2 => 2)/cpu/forward/NamedTuple/(2, 128) 1582 ns 1034.9936305732483 ns 1.53
Dense(2 => 2)/cpu/forward/ComponentArray/(2, 128) 1464.8 ns 1058.9415204678362 ns 1.38
Dense(2 => 2)/cpu/forward/Flux/(2, 128) 1802.0566037735848 ns 1797.0892857142858 ns 1.00
Dense(2 => 2)/cpu/forward/SimpleChains/(2, 128) 179.46036161335186 ns 180.6596638655462 ns 0.99
Dense(20 => 20)/cpu/reverse/ReverseDiff (compiled)/(20, 128) 17282 ns 17252 ns 1.00
Dense(20 => 20)/cpu/reverse/Zygote/(20, 128) 13104 ns 12543 ns 1.04
Dense(20 => 20)/cpu/reverse/Tracker/(20, 128) 37641 ns 39294 ns 0.96
Dense(20 => 20)/cpu/reverse/ReverseDiff/(20, 128) 29095 ns 28989.5 ns 1.00
Dense(20 => 20)/cpu/reverse/Flux/(20, 128) 21600 ns 20068 ns 1.08
Dense(20 => 20)/cpu/reverse/SimpleChains/(20, 128) 17072 ns 17102 ns 1.00
Dense(20 => 20)/cpu/reverse/Enzyme/(20, 128) 25698 ns 25538 ns 1.01
Dense(20 => 20)/cpu/forward/NamedTuple/(20, 128) 2609.3333333333335 ns 1482.7 ns 1.76
Dense(20 => 20)/cpu/forward/ComponentArray/(20, 128) 2913.222222222222 ns 1498.8 ns 1.94
Dense(20 => 20)/cpu/forward/Flux/(20, 128) 4874.857142857143 ns 4862 ns 1.00
Dense(20 => 20)/cpu/forward/SimpleChains/(20, 128) 1654.1 ns 1654.1 ns 1
Conv((3, 3), 3 => 3)/cpu/reverse/ReverseDiff (compiled)/(64, 64, 3, 128) 88901650 ns 84249197.5 ns 1.06
Conv((3, 3), 3 => 3)/cpu/reverse/Zygote/(64, 64, 3, 128) 76591182 ns 76847618 ns 1.00
Conv((3, 3), 3 => 3)/cpu/reverse/Tracker/(64, 64, 3, 128) 145816642 ns 155971883 ns 0.93
Conv((3, 3), 3 => 3)/cpu/reverse/ReverseDiff/(64, 64, 3, 128) 174645570 ns 171220777 ns 1.02
Conv((3, 3), 3 => 3)/cpu/reverse/Flux/(64, 64, 3, 128) 162467847 ns 163576307 ns 0.99
Conv((3, 3), 3 => 3)/cpu/reverse/SimpleChains/(64, 64, 3, 128) 11901058 ns 11997272 ns 0.99
Conv((3, 3), 3 => 3)/cpu/reverse/Enzyme/(64, 64, 3, 128) 195200307.5 ns 189232749 ns 1.03
Conv((3, 3), 3 => 3)/cpu/forward/NamedTuple/(64, 64, 3, 128) 15544763.5 ns 15985673.5 ns 0.97
Conv((3, 3), 3 => 3)/cpu/forward/ComponentArray/(64, 64, 3, 128) 15554299 ns 15990444 ns 0.97
Conv((3, 3), 3 => 3)/cpu/forward/Flux/(64, 64, 3, 128) 42589068.5 ns 26907120 ns 1.58
Conv((3, 3), 3 => 3)/cpu/forward/SimpleChains/(64, 64, 3, 128) 6385176 ns 6388864 ns 1.00
vgg16/cpu/reverse/Zygote/(32, 32, 3, 16) 1065642865.5 ns 1075154615 ns 0.99
vgg16/cpu/reverse/Zygote/(32, 32, 3, 64) 2933656388 ns 2991995826 ns 0.98
vgg16/cpu/reverse/Zygote/(32, 32, 3, 2) 169506774 ns 203143942 ns 0.83
vgg16/cpu/reverse/Tracker/(32, 32, 3, 16) 1395702878 ns 1327470294 ns 1.05
vgg16/cpu/reverse/Tracker/(32, 32, 3, 64) 3742059235 ns 3714520054 ns 1.01
vgg16/cpu/reverse/Tracker/(32, 32, 3, 2) 384902400.5 ns 393360752 ns 0.98
vgg16/cpu/reverse/Flux/(32, 32, 3, 16) 1673204957.5 ns 1513048001 ns 1.11
vgg16/cpu/reverse/Flux/(32, 32, 3, 64) 4683819068 ns 4465469598 ns 1.05
vgg16/cpu/reverse/Flux/(32, 32, 3, 2) 459579475 ns 427939919 ns 1.07
vgg16/cpu/forward/NamedTuple/(32, 32, 3, 16) 361855245 ns 373761436 ns 0.97
vgg16/cpu/forward/NamedTuple/(32, 32, 3, 64) 884689737 ns 904133788 ns 0.98
vgg16/cpu/forward/NamedTuple/(32, 32, 3, 2) 56182817 ns 51595239.5 ns 1.09
vgg16/cpu/forward/ComponentArray/(32, 32, 3, 16) 364122724 ns 374634761 ns 0.97
vgg16/cpu/forward/ComponentArray/(32, 32, 3, 64) 880222233.5 ns 915540600 ns 0.96
vgg16/cpu/forward/ComponentArray/(32, 32, 3, 2) 55637947 ns 50975700 ns 1.09
vgg16/cpu/forward/Flux/(32, 32, 3, 16) 562801142 ns 519866147 ns 1.08
vgg16/cpu/forward/Flux/(32, 32, 3, 64) 1517562171 ns 1395993770 ns 1.09
vgg16/cpu/forward/Flux/(32, 32, 3, 2) 171205515 ns 166463654 ns 1.03
Conv((3, 3), 64 => 64)/cpu/reverse/ReverseDiff (compiled)/(64, 64, 64, 128) 1254741222 ns 1256120343.5 ns 1.00
Conv((3, 3), 64 => 64)/cpu/reverse/Zygote/(64, 64, 64, 128) 1561957190.5 ns 1585667224.5 ns 0.99
Conv((3, 3), 64 => 64)/cpu/reverse/Tracker/(64, 64, 64, 128) 2274530908 ns 2358338032 ns 0.96
Conv((3, 3), 64 => 64)/cpu/reverse/ReverseDiff/(64, 64, 64, 128) 2467776572 ns 2629770018 ns 0.94
Conv((3, 3), 64 => 64)/cpu/reverse/Flux/(64, 64, 64, 128) 2236433184 ns 2299487877 ns 0.97
Conv((3, 3), 64 => 64)/cpu/reverse/Enzyme/(64, 64, 64, 128) 2124313017 ns 2202187002 ns 0.96
Conv((3, 3), 64 => 64)/cpu/forward/NamedTuple/(64, 64, 64, 128) 285138498 ns 288765459 ns 0.99
Conv((3, 3), 64 => 64)/cpu/forward/ComponentArray/(64, 64, 64, 128) 284816604 ns 287589792 ns 0.99
Conv((3, 3), 64 => 64)/cpu/forward/Flux/(64, 64, 64, 128) 426557087 ns 474169861.5 ns 0.90
Conv((3, 3), 1 => 1)/cpu/reverse/ReverseDiff (compiled)/(64, 64, 1, 128) 11831175 ns 11665649 ns 1.01
Conv((3, 3), 1 => 1)/cpu/reverse/Zygote/(64, 64, 1, 128) 34523223 ns 34793586 ns 0.99
Conv((3, 3), 1 => 1)/cpu/reverse/Tracker/(64, 64, 1, 128) 16447087 ns 16498583 ns 1.00
Conv((3, 3), 1 => 1)/cpu/reverse/ReverseDiff/(64, 64, 1, 128) 21042971 ns 21107303 ns 1.00
Conv((3, 3), 1 => 1)/cpu/reverse/Flux/(64, 64, 1, 128) 15151300 ns 15300235 ns 0.99
Conv((3, 3), 1 => 1)/cpu/reverse/SimpleChains/(64, 64, 1, 128) 1151983 ns 1163279.5 ns 0.99
Conv((3, 3), 1 => 1)/cpu/reverse/Enzyme/(64, 64, 1, 128) 35724522 ns 36078505 ns 0.99
Conv((3, 3), 1 => 1)/cpu/forward/NamedTuple/(64, 64, 1, 128) 4546295 ns 4714693.5 ns 0.96
Conv((3, 3), 1 => 1)/cpu/forward/ComponentArray/(64, 64, 1, 128) 4524386 ns 4687438 ns 0.97
Conv((3, 3), 1 => 1)/cpu/forward/Flux/(64, 64, 1, 128) 1970497 ns 1991858.5 ns 0.99
Conv((3, 3), 1 => 1)/cpu/forward/SimpleChains/(64, 64, 1, 128) 197294 ns 198611 ns 0.99
Dense(200 => 200)/cpu/reverse/ReverseDiff (compiled)/(200, 128) 380051 ns 379980 ns 1.00
Dense(200 => 200)/cpu/reverse/Zygote/(200, 128) 222526 ns 213118.5 ns 1.04
Dense(200 => 200)/cpu/reverse/Tracker/(200, 128) 384974.5 ns 388676 ns 0.99
Dense(200 => 200)/cpu/reverse/ReverseDiff/(200, 128) 528056 ns 512398 ns 1.03
Dense(200 => 200)/cpu/reverse/Flux/(200, 128) 294290 ns 293543.5 ns 1.00
Dense(200 => 200)/cpu/reverse/SimpleChains/(200, 128) 408554 ns 407692 ns 1.00
Dense(200 => 200)/cpu/reverse/Enzyme/(200, 128) 429642.5 ns 428922 ns 1.00
Dense(200 => 200)/cpu/forward/NamedTuple/(200, 128) 70512 ns 57699 ns 1.22
Dense(200 => 200)/cpu/forward/ComponentArray/(200, 128) 70061 ns 60272 ns 1.16
Dense(200 => 200)/cpu/forward/Flux/(200, 128) 92674 ns 92693 ns 1.00
Dense(200 => 200)/cpu/forward/SimpleChains/(200, 128) 104425 ns 104605 ns 1.00
Conv((3, 3), 16 => 16)/cpu/reverse/ReverseDiff (compiled)/(64, 64, 16, 128) 305601445 ns 336588187 ns 0.91
Conv((3, 3), 16 => 16)/cpu/reverse/Zygote/(64, 64, 16, 128) 288797620.5 ns 286481480.5 ns 1.01
Conv((3, 3), 16 => 16)/cpu/reverse/Tracker/(64, 64, 16, 128) 544934785.5 ns 553202812.5 ns 0.99
Conv((3, 3), 16 => 16)/cpu/reverse/ReverseDiff/(64, 64, 16, 128) 589046818 ns 631696024 ns 0.93
Conv((3, 3), 16 => 16)/cpu/reverse/Flux/(64, 64, 16, 128) 552663074 ns 584369049 ns 0.95
Conv((3, 3), 16 => 16)/cpu/reverse/SimpleChains/(64, 64, 16, 128) 319651032.5 ns 332985122 ns 0.96
Conv((3, 3), 16 => 16)/cpu/reverse/Enzyme/(64, 64, 16, 128) 660000513 ns 608969909 ns 1.08
Conv((3, 3), 16 => 16)/cpu/forward/NamedTuple/(64, 64, 16, 128) 39489513.5 ns 40188939 ns 0.98
Conv((3, 3), 16 => 16)/cpu/forward/ComponentArray/(64, 64, 16, 128) 39313659 ns 40059867 ns 0.98
Conv((3, 3), 16 => 16)/cpu/forward/Flux/(64, 64, 16, 128) 99796304 ns 109125318.5 ns 0.91
Conv((3, 3), 16 => 16)/cpu/forward/SimpleChains/(64, 64, 16, 128) 28394041 ns 28648755 ns 0.99
Dense(2000 => 2000)/cpu/reverse/ReverseDiff (compiled)/(2000, 128) 21157343.5 ns 21170151.5 ns 1.00
Dense(2000 => 2000)/cpu/reverse/Zygote/(2000, 128) 16678967 ns 21223422 ns 0.79
Dense(2000 => 2000)/cpu/reverse/Tracker/(2000, 128) 22876715 ns 22980720 ns 1.00
Dense(2000 => 2000)/cpu/reverse/ReverseDiff/(2000, 128) 28080465 ns 26450851 ns 1.06
Dense(2000 => 2000)/cpu/reverse/Flux/(2000, 128) 19385408 ns 19326108.5 ns 1.00
Dense(2000 => 2000)/cpu/reverse/Enzyme/(2000, 128) 20728329 ns 20888920 ns 0.99
Dense(2000 => 2000)/cpu/forward/NamedTuple/(2000, 128) 5545671.5 ns 6116570 ns 0.91
Dense(2000 => 2000)/cpu/forward/ComponentArray/(2000, 128) 5626649 ns 6384401 ns 0.88
Dense(2000 => 2000)/cpu/forward/Flux/(2000, 128) 6521885 ns 6570749 ns 0.99

This comment was automatically generated by workflow using github-action-benchmark.

Please sign in to comment.