[do not merge] Test hand-coded sum on kron and kron_view

[yebai]

No description provided.

[codecov]

Codecov Report

All modified and coverable lines are covered by tests ✅

see 3 files with indirect coverage changes

[github-actions]

Performance Ratio:
Ratio of time to compute gradient and time to compute function.
Warning: results are very approximate! See here for more context.

┌────────────────────────────┬──────────┬──────────┬─────────────┬─────────┐
│                      Label │ Mooncake │   Zygote │ ReverseDiff │  Enzyme │
│                     String │   String │   String │      String │  String │
├────────────────────────────┼──────────┼──────────┼─────────────┼─────────┤
│                   sum_1000 │     83.2 │      1.1 │        5.51 │    8.21 │
│                  _sum_1000 │     7.11 │   1660.0 │        33.6 │    1.07 │
│               sum_sin_1000 │     2.29 │     1.69 │        10.7 │    1.99 │
│              _sum_sin_1000 │     2.81 │    338.0 │        13.2 │    2.43 │
│                   kron_sum │     25.5 │ 779000.0 │        84.8 │    3.38 │
│              kron_view_sum │     12.7 │ 435000.0 │        53.9 │    22.0 │
│      naive_map_sin_cos_exp │     2.54 │  missing │        7.18 │    2.32 │
│            map_sin_cos_exp │     2.67 │     1.44 │        5.93 │    2.86 │
│      broadcast_sin_cos_exp │      2.6 │     2.27 │        1.45 │    2.25 │
│                 simple_mlp │     4.95 │     3.07 │        7.16 │    3.44 │
│                     gp_lml │     4.55 │     2.22 │     missing │    2.99 │
│ turing_broadcast_benchmark │     3.15 │  missing │        25.2 │ missing │
│         large_single_block │     4.42 │   4760.0 │        31.0 │    2.18 │
└────────────────────────────┴──────────┴──────────┴─────────────┴─────────┘