-
Notifications
You must be signed in to change notification settings - Fork 59
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
11 changed files
with
29 additions
and
31 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
937efad
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Benchmark Results
Dense(2 => 2)/cpu/reverse/ReverseDiff (compiled)/(2, 128)
3669.375
ns3889.75
ns0.94
Dense(2 => 2)/cpu/reverse/Zygote/(2, 128)
7200
ns7213.428571428572
ns1.00
Dense(2 => 2)/cpu/reverse/Tracker/(2, 128)
21054.5
ns20608
ns1.02
Dense(2 => 2)/cpu/reverse/ReverseDiff/(2, 128)
9781.3
ns9816.3
ns1.00
Dense(2 => 2)/cpu/reverse/Flux/(2, 128)
8908.6
ns9185.2
ns0.97
Dense(2 => 2)/cpu/reverse/SimpleChains/(2, 128)
4449.5
ns4510.875
ns0.99
Dense(2 => 2)/cpu/reverse/Enzyme/(2, 128)
1157.9930555555557
ns1166.159420289855
ns0.99
Dense(2 => 2)/cpu/forward/NamedTuple/(2, 128)
1171.8985507246377
ns1111.1946308724832
ns1.05
Dense(2 => 2)/cpu/forward/ComponentArray/(2, 128)
1185.8125
ns1187.955223880597
ns1.00
Dense(2 => 2)/cpu/forward/Flux/(2, 128)
1791.6440677966102
ns1795.1666666666667
ns1.00
Dense(2 => 2)/cpu/forward/SimpleChains/(2, 128)
180.02920723226703
ns179.85854341736695
ns1.00
Dense(20 => 20)/cpu/reverse/ReverseDiff (compiled)/(20, 128)
17283
ns17242
ns1.00
Dense(20 => 20)/cpu/reverse/Zygote/(20, 128)
16841
ns16641
ns1.01
Dense(20 => 20)/cpu/reverse/Tracker/(20, 128)
37390
ns36418
ns1.03
Dense(20 => 20)/cpu/reverse/ReverseDiff/(20, 128)
28994
ns28994
ns1
Dense(20 => 20)/cpu/reverse/Flux/(20, 128)
20027
ns21290
ns0.94
Dense(20 => 20)/cpu/reverse/SimpleChains/(20, 128)
17182
ns17383
ns0.99
Dense(20 => 20)/cpu/reverse/Enzyme/(20, 128)
4328
ns4282.285714285715
ns1.01
Dense(20 => 20)/cpu/forward/NamedTuple/(20, 128)
3870.875
ns3815.875
ns1.01
Dense(20 => 20)/cpu/forward/ComponentArray/(20, 128)
3954.875
ns3921
ns1.01
Dense(20 => 20)/cpu/forward/Flux/(20, 128)
4847.642857142857
ns4816.142857142857
ns1.01
Dense(20 => 20)/cpu/forward/SimpleChains/(20, 128)
1658.1
ns1651.1
ns1.00
Conv((3, 3), 3 => 3)/cpu/reverse/ReverseDiff (compiled)/(64, 64, 3, 128)
40009085
ns38826337.5
ns1.03
Conv((3, 3), 3 => 3)/cpu/reverse/Zygote/(64, 64, 3, 128)
58044866
ns57888052
ns1.00
Conv((3, 3), 3 => 3)/cpu/reverse/Tracker/(64, 64, 3, 128)
81652371
ns75523456
ns1.08
Conv((3, 3), 3 => 3)/cpu/reverse/ReverseDiff/(64, 64, 3, 128)
84691568
ns88301980
ns0.96
Conv((3, 3), 3 => 3)/cpu/reverse/Flux/(64, 64, 3, 128)
75567907
ns72229490
ns1.05
Conv((3, 3), 3 => 3)/cpu/reverse/SimpleChains/(64, 64, 3, 128)
11702113
ns11634194
ns1.01
Conv((3, 3), 3 => 3)/cpu/reverse/Enzyme/(64, 64, 3, 128)
6962047.5
ns6911425
ns1.01
Conv((3, 3), 3 => 3)/cpu/forward/NamedTuple/(64, 64, 3, 128)
7131554
ns7082578.5
ns1.01
Conv((3, 3), 3 => 3)/cpu/forward/ComponentArray/(64, 64, 3, 128)
7068401
ns6994755
ns1.01
Conv((3, 3), 3 => 3)/cpu/forward/Flux/(64, 64, 3, 128)
12275161.5
ns9789325
ns1.25
Conv((3, 3), 3 => 3)/cpu/forward/SimpleChains/(64, 64, 3, 128)
6382810.5
ns6381274
ns1.00
vgg16/cpu/reverse/Zygote/(32, 32, 3, 16)
714086754
ns706120562
ns1.01
vgg16/cpu/reverse/Zygote/(32, 32, 3, 64)
2579434858
ns2529871556
ns1.02
vgg16/cpu/reverse/Zygote/(32, 32, 3, 2)
148383143
ns131797096
ns1.13
vgg16/cpu/reverse/Tracker/(32, 32, 3, 16)
891741497
ns757974733
ns1.18
vgg16/cpu/reverse/Tracker/(32, 32, 3, 64)
3088439700
ns2932092144
ns1.05
vgg16/cpu/reverse/Tracker/(32, 32, 3, 2)
212058426
ns189094496
ns1.12
vgg16/cpu/reverse/Flux/(32, 32, 3, 16)
653749242
ns663524511.5
ns0.99
vgg16/cpu/reverse/Flux/(32, 32, 3, 64)
2642538349
ns2577224378.5
ns1.03
vgg16/cpu/reverse/Flux/(32, 32, 3, 2)
128699402
ns125424056
ns1.03
vgg16/cpu/forward/NamedTuple/(32, 32, 3, 16)
175013829.5
ns174066006
ns1.01
vgg16/cpu/forward/NamedTuple/(32, 32, 3, 64)
654368735
ns653654001
ns1.00
vgg16/cpu/forward/NamedTuple/(32, 32, 3, 2)
45671588
ns34513878
ns1.32
vgg16/cpu/forward/ComponentArray/(32, 32, 3, 16)
165691299
ns164808835
ns1.01
vgg16/cpu/forward/ComponentArray/(32, 32, 3, 64)
644989507
ns634945112.5
ns1.02
vgg16/cpu/forward/ComponentArray/(32, 32, 3, 2)
45474454
ns29766865
ns1.53
vgg16/cpu/forward/Flux/(32, 32, 3, 16)
227248905
ns187595950
ns1.21
vgg16/cpu/forward/Flux/(32, 32, 3, 64)
775116259
ns722751963
ns1.07
vgg16/cpu/forward/Flux/(32, 32, 3, 2)
40575003.5
ns37544548
ns1.08
Conv((3, 3), 64 => 64)/cpu/reverse/ReverseDiff (compiled)/(64, 64, 64, 128)
1266387049.5
ns1226815927.5
ns1.03
Conv((3, 3), 64 => 64)/cpu/reverse/Zygote/(64, 64, 64, 128)
1873245890
ns1859890354
ns1.01
Conv((3, 3), 64 => 64)/cpu/reverse/Tracker/(64, 64, 64, 128)
2337050962
ns2339835190
ns1.00
Conv((3, 3), 64 => 64)/cpu/reverse/ReverseDiff/(64, 64, 64, 128)
2598176961
ns2525615729
ns1.03
Conv((3, 3), 64 => 64)/cpu/reverse/Flux/(64, 64, 64, 128)
1906864622.5
ns1865477869.5
ns1.02
Conv((3, 3), 64 => 64)/cpu/reverse/Enzyme/(64, 64, 64, 128)
323660356.5
ns322877161
ns1.00
Conv((3, 3), 64 => 64)/cpu/forward/NamedTuple/(64, 64, 64, 128)
323114250
ns320721263
ns1.01
Conv((3, 3), 64 => 64)/cpu/forward/ComponentArray/(64, 64, 64, 128)
319854782
ns317208348
ns1.01
Conv((3, 3), 64 => 64)/cpu/forward/Flux/(64, 64, 64, 128)
431236413
ns418779532
ns1.03
Conv((3, 3), 1 => 1)/cpu/reverse/ReverseDiff (compiled)/(64, 64, 1, 128)
11856906
ns11864932
ns1.00
Conv((3, 3), 1 => 1)/cpu/reverse/Zygote/(64, 64, 1, 128)
17995791
ns17886839.5
ns1.01
Conv((3, 3), 1 => 1)/cpu/reverse/Tracker/(64, 64, 1, 128)
19308830
ns19045196
ns1.01
Conv((3, 3), 1 => 1)/cpu/reverse/ReverseDiff/(64, 64, 1, 128)
24002610
ns23830079.5
ns1.01
Conv((3, 3), 1 => 1)/cpu/reverse/Flux/(64, 64, 1, 128)
18021078
ns17899726
ns1.01
Conv((3, 3), 1 => 1)/cpu/reverse/SimpleChains/(64, 64, 1, 128)
1168729
ns1148335
ns1.02
Conv((3, 3), 1 => 1)/cpu/reverse/Enzyme/(64, 64, 1, 128)
2067715
ns2052685
ns1.01
Conv((3, 3), 1 => 1)/cpu/forward/NamedTuple/(64, 64, 1, 128)
2082044
ns2063356
ns1.01
Conv((3, 3), 1 => 1)/cpu/forward/ComponentArray/(64, 64, 1, 128)
2083795
ns2066360
ns1.01
Conv((3, 3), 1 => 1)/cpu/forward/Flux/(64, 64, 1, 128)
2078152
ns2051643
ns1.01
Conv((3, 3), 1 => 1)/cpu/forward/SimpleChains/(64, 64, 1, 128)
198179
ns196415
ns1.01
Dense(200 => 200)/cpu/reverse/ReverseDiff (compiled)/(200, 128)
294960
ns289318
ns1.02
Dense(200 => 200)/cpu/reverse/Zygote/(200, 128)
268370
ns263239
ns1.02
Dense(200 => 200)/cpu/reverse/Tracker/(200, 128)
372856
ns359839
ns1.04
Dense(200 => 200)/cpu/reverse/ReverseDiff/(200, 128)
414518.5
ns405253
ns1.02
Dense(200 => 200)/cpu/reverse/Flux/(200, 128)
278368
ns272277
ns1.02
Dense(200 => 200)/cpu/reverse/SimpleChains/(200, 128)
410746
ns405564
ns1.01
Dense(200 => 200)/cpu/reverse/Enzyme/(200, 128)
83776
ns83124
ns1.01
Dense(200 => 200)/cpu/forward/NamedTuple/(200, 128)
81953
ns80740
ns1.02
Dense(200 => 200)/cpu/forward/ComponentArray/(200, 128)
83195
ns82383
ns1.01
Dense(200 => 200)/cpu/forward/Flux/(200, 128)
87764
ns86060
ns1.02
Dense(200 => 200)/cpu/forward/SimpleChains/(200, 128)
104685
ns104325
ns1.00
Conv((3, 3), 16 => 16)/cpu/reverse/ReverseDiff (compiled)/(64, 64, 16, 128)
198244459
ns197459994
ns1.00
Conv((3, 3), 16 => 16)/cpu/reverse/Zygote/(64, 64, 16, 128)
329497937.5
ns327332523.5
ns1.01
Conv((3, 3), 16 => 16)/cpu/reverse/Tracker/(64, 64, 16, 128)
407311865.5
ns367989344
ns1.11
Conv((3, 3), 16 => 16)/cpu/reverse/ReverseDiff/(64, 64, 16, 128)
452728713
ns439797244.5
ns1.03
Conv((3, 3), 16 => 16)/cpu/reverse/Flux/(64, 64, 16, 128)
388579645
ns374931694
ns1.04
Conv((3, 3), 16 => 16)/cpu/reverse/SimpleChains/(64, 64, 16, 128)
319868232.5
ns326044951
ns0.98
Conv((3, 3), 16 => 16)/cpu/reverse/Enzyme/(64, 64, 16, 128)
44289039
ns44053216
ns1.01
Conv((3, 3), 16 => 16)/cpu/forward/NamedTuple/(64, 64, 16, 128)
44316868
ns44122534.5
ns1.00
Conv((3, 3), 16 => 16)/cpu/forward/ComponentArray/(64, 64, 16, 128)
43960351
ns43811761
ns1.00
Conv((3, 3), 16 => 16)/cpu/forward/Flux/(64, 64, 16, 128)
51507810
ns53017284.5
ns0.97
Conv((3, 3), 16 => 16)/cpu/forward/SimpleChains/(64, 64, 16, 128)
28435005
ns27849167
ns1.02
Dense(2000 => 2000)/cpu/reverse/ReverseDiff (compiled)/(2000, 128)
18968878
ns18739042.5
ns1.01
Dense(2000 => 2000)/cpu/reverse/Zygote/(2000, 128)
19614768
ns19499913
ns1.01
Dense(2000 => 2000)/cpu/reverse/Tracker/(2000, 128)
23514329
ns23187751
ns1.01
Dense(2000 => 2000)/cpu/reverse/ReverseDiff/(2000, 128)
24219437.5
ns24095441
ns1.01
Dense(2000 => 2000)/cpu/reverse/Flux/(2000, 128)
19657817
ns19512675
ns1.01
Dense(2000 => 2000)/cpu/reverse/Enzyme/(2000, 128)
6533333
ns6502673
ns1.00
Dense(2000 => 2000)/cpu/forward/NamedTuple/(2000, 128)
6543909
ns6533912
ns1.00
Dense(2000 => 2000)/cpu/forward/ComponentArray/(2000, 128)
6522365
ns6481314
ns1.01
Dense(2000 => 2000)/cpu/forward/Flux/(2000, 128)
6506154.5
ns6489178
ns1.00
This comment was automatically generated by workflow using github-action-benchmark.