Skip to content

Commit

Permalink
chore: use @return in tutorials
Browse files Browse the repository at this point in the history
  • Loading branch information
@avik-pal
avik-pal committed Jul 12, 2024
1 parent b50646a commit 6a9ef65
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions examples/SymbolicOptimalControl/main.jl
View file
Original file line number Diff line number Diff line change
Expand Up @@ -81,8 +81,8 @@ function construct_ude(mlp, solver; kwargs...)
sensealg=QuadratureAdjoint(; autojacvec=ReverseDiffVJP(true)), kwargs...)

us = mlp(reshape(ts, 1, :), ps.mlp)
ret_sol === Val(true) && return sol, us
return Array(sol), us
ret_sol === Val(true) && @return sol, us
@return Array(sol), us
end
end

Expand Down

1 comment on commit 6a9ef65

@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: 6a9ef65 Previous: c0f03a0 Ratio
Dense(2 => 2)/cpu/reverse/ReverseDiff (compiled)/(2, 128) 3705.75 ns 3604.25 ns 1.03
Dense(2 => 2)/cpu/reverse/Zygote/(2, 128) 7107.4 ns 7170 ns 0.99
Dense(2 => 2)/cpu/reverse/Tracker/(2, 128) 20799 ns 20669 ns 1.01
Dense(2 => 2)/cpu/reverse/ReverseDiff/(2, 128) 9710.3 ns 9613.8 ns 1.01
Dense(2 => 2)/cpu/reverse/Flux/(2, 128) 9047 ns 9197 ns 0.98
Dense(2 => 2)/cpu/reverse/SimpleChains/(2, 128) 4463.375 ns 4430.75 ns 1.01
Dense(2 => 2)/cpu/reverse/Enzyme/(2, 128) 1160.731884057971 ns 1151.2978723404256 ns 1.01
Dense(2 => 2)/cpu/forward/NamedTuple/(2, 128) 1111.9610389610389 ns 1166.2941176470588 ns 0.95
Dense(2 => 2)/cpu/forward/ComponentArray/(2, 128) 1187.8854961832062 ns 1166.7202797202797 ns 1.02
Dense(2 => 2)/cpu/forward/Flux/(2, 128) 1791.5333333333333 ns 1769.4193548387098 ns 1.01
Dense(2 => 2)/cpu/forward/SimpleChains/(2, 128) 179.46262341325811 ns 179.31849791376914 ns 1.00
Dense(20 => 20)/cpu/reverse/ReverseDiff (compiled)/(20, 128) 17282 ns 17353 ns 1.00
Dense(20 => 20)/cpu/reverse/Zygote/(20, 128) 16862 ns 16601 ns 1.02
Dense(20 => 20)/cpu/reverse/Tracker/(20, 128) 37170 ns 36883.5 ns 1.01
Dense(20 => 20)/cpu/reverse/ReverseDiff/(20, 128) 29185 ns 27962 ns 1.04
Dense(20 => 20)/cpu/reverse/Flux/(20, 128) 20228 ns 20127 ns 1.01
Dense(20 => 20)/cpu/reverse/SimpleChains/(20, 128) 17303 ns 17012 ns 1.02
Dense(20 => 20)/cpu/reverse/Enzyme/(20, 128) 4306.714285714285 ns 4306.571428571428 ns 1.00
Dense(20 => 20)/cpu/forward/NamedTuple/(20, 128) 3868.5 ns 3825.875 ns 1.01
Dense(20 => 20)/cpu/forward/ComponentArray/(20, 128) 3936.25 ns 3911 ns 1.01
Dense(20 => 20)/cpu/forward/Flux/(20, 128) 4992.142857142857 ns 4856.285714285715 ns 1.03
Dense(20 => 20)/cpu/forward/SimpleChains/(20, 128) 1654.1 ns 1653.1 ns 1.00
Conv((3, 3), 3 => 3)/cpu/reverse/ReverseDiff (compiled)/(64, 64, 3, 128) 48611821 ns 49241683 ns 0.99
Conv((3, 3), 3 => 3)/cpu/reverse/Zygote/(64, 64, 3, 128) 57919354 ns 57960280 ns 1.00
Conv((3, 3), 3 => 3)/cpu/reverse/Tracker/(64, 64, 3, 128) 109926255 ns 110687528 ns 0.99
Conv((3, 3), 3 => 3)/cpu/reverse/ReverseDiff/(64, 64, 3, 128) 107281465 ns 107034221 ns 1.00
Conv((3, 3), 3 => 3)/cpu/reverse/Flux/(64, 64, 3, 128) 91622516 ns 107263120.5 ns 0.85
Conv((3, 3), 3 => 3)/cpu/reverse/SimpleChains/(64, 64, 3, 128) 11726802 ns 11972413 ns 0.98
Conv((3, 3), 3 => 3)/cpu/reverse/Enzyme/(64, 64, 3, 128) 6972718 ns 6920385 ns 1.01
Conv((3, 3), 3 => 3)/cpu/forward/NamedTuple/(64, 64, 3, 128) 7145941 ns 7063697 ns 1.01
Conv((3, 3), 3 => 3)/cpu/forward/ComponentArray/(64, 64, 3, 128) 7086834.5 ns 7116659 ns 1.00
Conv((3, 3), 3 => 3)/cpu/forward/Flux/(64, 64, 3, 128) 18155782 ns 19130332 ns 0.95
Conv((3, 3), 3 => 3)/cpu/forward/SimpleChains/(64, 64, 3, 128) 6400164 ns 6378626.5 ns 1.00
vgg16/cpu/reverse/Zygote/(32, 32, 3, 16) 695663529 ns 676277931 ns 1.03
vgg16/cpu/reverse/Zygote/(32, 32, 3, 64) 2551834000 ns 2529085990 ns 1.01
vgg16/cpu/reverse/Zygote/(32, 32, 3, 2) 144828075.5 ns 133234444 ns 1.09
vgg16/cpu/reverse/Tracker/(32, 32, 3, 16) 937571020 ns 944154886 ns 0.99
vgg16/cpu/reverse/Tracker/(32, 32, 3, 64) 3760657273 ns 3667947313 ns 1.03
vgg16/cpu/reverse/Tracker/(32, 32, 3, 2) 218456733 ns 228784367 ns 0.95
vgg16/cpu/reverse/Flux/(32, 32, 3, 16) 1023397506 ns 999814712.5 ns 1.02
vgg16/cpu/reverse/Flux/(32, 32, 3, 64) 2833424793 ns 3305914576 ns 0.86
vgg16/cpu/reverse/Flux/(32, 32, 3, 2) 135152975 ns 151210631.5 ns 0.89
vgg16/cpu/forward/NamedTuple/(32, 32, 3, 16) 173677436.5 ns 171977811.5 ns 1.01
vgg16/cpu/forward/NamedTuple/(32, 32, 3, 64) 667093038.5 ns 648927758 ns 1.03
vgg16/cpu/forward/NamedTuple/(32, 32, 3, 2) 35692896 ns 34417243 ns 1.04
vgg16/cpu/forward/ComponentArray/(32, 32, 3, 16) 165432458 ns 168092289 ns 0.98
vgg16/cpu/forward/ComponentArray/(32, 32, 3, 64) 638281808 ns 640573982 ns 1.00
vgg16/cpu/forward/ComponentArray/(32, 32, 3, 2) 30282180 ns 29659101 ns 1.02
vgg16/cpu/forward/Flux/(32, 32, 3, 16) 228436072 ns 200849870 ns 1.14
vgg16/cpu/forward/Flux/(32, 32, 3, 64) 907887712 ns 833750681 ns 1.09
vgg16/cpu/forward/Flux/(32, 32, 3, 2) 37665017.5 ns 37541910 ns 1.00
Conv((3, 3), 64 => 64)/cpu/reverse/ReverseDiff (compiled)/(64, 64, 64, 128) 1202743851 ns 1179860972 ns 1.02
Conv((3, 3), 64 => 64)/cpu/reverse/Zygote/(64, 64, 64, 128) 1871595149.5 ns 1861619729 ns 1.01
Conv((3, 3), 64 => 64)/cpu/reverse/Tracker/(64, 64, 64, 128) 2375315336 ns 2504278030 ns 0.95
Conv((3, 3), 64 => 64)/cpu/reverse/ReverseDiff/(64, 64, 64, 128) 2568230230 ns 2474701932 ns 1.04
Conv((3, 3), 64 => 64)/cpu/reverse/Flux/(64, 64, 64, 128) 1850936169 ns 1999749817.5 ns 0.93
Conv((3, 3), 64 => 64)/cpu/reverse/Enzyme/(64, 64, 64, 128) 325578942 ns 318275806.5 ns 1.02
Conv((3, 3), 64 => 64)/cpu/forward/NamedTuple/(64, 64, 64, 128) 325502546 ns 317826145 ns 1.02
Conv((3, 3), 64 => 64)/cpu/forward/ComponentArray/(64, 64, 64, 128) 327773885 ns 316745015 ns 1.03
Conv((3, 3), 64 => 64)/cpu/forward/Flux/(64, 64, 64, 128) 362130805 ns 429906704 ns 0.84
Conv((3, 3), 1 => 1)/cpu/reverse/ReverseDiff (compiled)/(64, 64, 1, 128) 11798402 ns 11667810 ns 1.01
Conv((3, 3), 1 => 1)/cpu/reverse/Zygote/(64, 64, 1, 128) 17952895 ns 17908951 ns 1.00
Conv((3, 3), 1 => 1)/cpu/reverse/Tracker/(64, 64, 1, 128) 19111937 ns 19106808 ns 1.00
Conv((3, 3), 1 => 1)/cpu/reverse/ReverseDiff/(64, 64, 1, 128) 23813592 ns 23766079 ns 1.00
Conv((3, 3), 1 => 1)/cpu/reverse/Flux/(64, 64, 1, 128) 17994614 ns 17918013 ns 1.00
Conv((3, 3), 1 => 1)/cpu/reverse/SimpleChains/(64, 64, 1, 128) 1161846 ns 1158548 ns 1.00
Conv((3, 3), 1 => 1)/cpu/reverse/Enzyme/(64, 64, 1, 128) 2071410 ns 2057538.5 ns 1.01
Conv((3, 3), 1 => 1)/cpu/forward/NamedTuple/(64, 64, 1, 128) 2088598.5 ns 2073008.5 ns 1.01
Conv((3, 3), 1 => 1)/cpu/forward/ComponentArray/(64, 64, 1, 128) 2095050 ns 2075418 ns 1.01
Conv((3, 3), 1 => 1)/cpu/forward/Flux/(64, 64, 1, 128) 2087753 ns 2076144 ns 1.01
Conv((3, 3), 1 => 1)/cpu/forward/SimpleChains/(64, 64, 1, 128) 204526 ns 199922.5 ns 1.02
Dense(200 => 200)/cpu/reverse/ReverseDiff (compiled)/(200, 128) 294988 ns 290632 ns 1.01
Dense(200 => 200)/cpu/reverse/Zygote/(200, 128) 266894 ns 263000 ns 1.01
Dense(200 => 200)/cpu/reverse/Tracker/(200, 128) 370339.5 ns 362270 ns 1.02
Dense(200 => 200)/cpu/reverse/ReverseDiff/(200, 128) 410705 ns 404589.5 ns 1.02
Dense(200 => 200)/cpu/reverse/Flux/(200, 128) 277925 ns 272262.5 ns 1.02
Dense(200 => 200)/cpu/reverse/SimpleChains/(200, 128) 409543 ns 407625 ns 1.00
Dense(200 => 200)/cpu/reverse/Enzyme/(200, 128) 83628 ns 83105 ns 1.01
Dense(200 => 200)/cpu/forward/NamedTuple/(200, 128) 82105 ns 80730 ns 1.02
Dense(200 => 200)/cpu/forward/ComponentArray/(200, 128) 83929 ns 81021 ns 1.04
Dense(200 => 200)/cpu/forward/Flux/(200, 128) 87756 ns 86121 ns 1.02
Dense(200 => 200)/cpu/forward/SimpleChains/(200, 128) 104467 ns 104365 ns 1.00
Conv((3, 3), 16 => 16)/cpu/reverse/ReverseDiff (compiled)/(64, 64, 16, 128) 190419837 ns 212248406.5 ns 0.90
Conv((3, 3), 16 => 16)/cpu/reverse/Zygote/(64, 64, 16, 128) 328138588.5 ns 327280336.5 ns 1.00
Conv((3, 3), 16 => 16)/cpu/reverse/Tracker/(64, 64, 16, 128) 383045958 ns 450238435.5 ns 0.85
Conv((3, 3), 16 => 16)/cpu/reverse/ReverseDiff/(64, 64, 16, 128) 460735044.5 ns 508490953 ns 0.91
Conv((3, 3), 16 => 16)/cpu/reverse/Flux/(64, 64, 16, 128) 380286964 ns 419359652 ns 0.91
Conv((3, 3), 16 => 16)/cpu/reverse/SimpleChains/(64, 64, 16, 128) 334237556 ns 317806942.5 ns 1.05
Conv((3, 3), 16 => 16)/cpu/reverse/Enzyme/(64, 64, 16, 128) 44410897 ns 44054551 ns 1.01
Conv((3, 3), 16 => 16)/cpu/forward/NamedTuple/(64, 64, 16, 128) 44630099.5 ns 44114540.5 ns 1.01
Conv((3, 3), 16 => 16)/cpu/forward/ComponentArray/(64, 64, 16, 128) 44306468 ns 43922579 ns 1.01
Conv((3, 3), 16 => 16)/cpu/forward/Flux/(64, 64, 16, 128) 50448519 ns 57543157 ns 0.88
Conv((3, 3), 16 => 16)/cpu/forward/SimpleChains/(64, 64, 16, 128) 28143904 ns 28648610 ns 0.98
Dense(2000 => 2000)/cpu/reverse/ReverseDiff (compiled)/(2000, 128) 19485606 ns 18727776 ns 1.04
Dense(2000 => 2000)/cpu/reverse/Zygote/(2000, 128) 19747168 ns 19448960.5 ns 1.02
Dense(2000 => 2000)/cpu/reverse/Tracker/(2000, 128) 23603854 ns 23160761 ns 1.02
Dense(2000 => 2000)/cpu/reverse/ReverseDiff/(2000, 128) 24284477.5 ns 24009458 ns 1.01
Dense(2000 => 2000)/cpu/reverse/Flux/(2000, 128) 19844857.5 ns 19509547 ns 1.02
Dense(2000 => 2000)/cpu/reverse/Enzyme/(2000, 128) 6534898 ns 6512284.5 ns 1.00
Dense(2000 => 2000)/cpu/forward/NamedTuple/(2000, 128) 6565242 ns 6510712 ns 1.01
Dense(2000 => 2000)/cpu/forward/ComponentArray/(2000, 128) 6551615 ns 6463513.5 ns 1.01
Dense(2000 => 2000)/cpu/forward/Flux/(2000, 128) 6522759 ns 6488277.5 ns 1.01

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

Please sign in to comment.