Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Manually fuse reindexing intermediates with blockwise reduction for cohorts. #300

Merged
merged 2 commits into from
May 2, 2024

Commits on May 2, 2024

  1. Manually fuse reindexing intermediates with blockwise reduction for c…

    …ohorts.
    
    ```
    | Change   | Before [627bf2b] <main>   | After [9d710529] <optimize-cohorts-graph>   |   Ratio | Benchmark (Parameter)                           |
    |----------|----------------------------|---------------------------------------------|---------|-------------------------------------------------|
    | -        | 3.39±0.02ms                | 2.98±0.01ms                                 |    0.88 | cohorts.PerfectMonthly.time_graph_construct     |
    | -        | 20                         | 17                                          |    0.85 | cohorts.PerfectMonthly.track_num_layers         |
    | -        | 23.0±0.07ms                | 19.0±0.1ms                                  |    0.83 | cohorts.ERA5Google.time_graph_construct         |
    | -        | 4878                       | 3978                                        |    0.82 | cohorts.ERA5Google.track_num_tasks              |
    | -        | 179±0.8ms                  | 147±0.5ms                                   |    0.82 | cohorts.OISST.time_graph_construct              |
    | -        | 159                        | 128                                         |    0.81 | cohorts.ERA5Google.track_num_layers             |
    | -        | 936                        | 762                                         |    0.81 | cohorts.PerfectMonthly.track_num_tasks          |
    | -        | 1221                       | 978                                         |    0.8  | cohorts.OISST.track_num_layers                  |
    | -        | 4929                       | 3834                                        |    0.78 | cohorts.ERA5DayOfYear.track_num_tasks           |
    | -        | 351                        | 274                                         |    0.78 | cohorts.NWMMidwest.track_num_layers             |
    | -        | 4562                       | 3468                                        |    0.76 | cohorts.ERA5DayOfYear.track_num_tasks_optimized |
    | -        | 164±1ms                    | 118±0.4ms                                   |    0.72 | cohorts.ERA5DayOfYear.time_graph_construct      |
    | -        | 1100                       | 735                                         |    0.67 | cohorts.ERA5DayOfYear.track_num_layers          |
    | -        | 3930                       | 2605                                        |    0.66 | cohorts.NWMMidwest.track_num_tasks              |
    | -        | 3715                       | 2409                                        |    0.65 | cohorts.NWMMidwest.track_num_tasks_optimized    |
    | -        | 28952                      | 18798                                       |    0.65 | cohorts.OISST.track_num_tasks                   |
    | -        | 27010                      | 16858                                       |    0.62 | cohorts.OISST.track_num_tasks_optimized         |
    ```
    dcherian committed May 2, 2024
    Configuration menu
    Copy the full SHA
    d5cb68c View commit details
    Browse the repository at this point in the history
  2. fix typing

    dcherian committed May 2, 2024
    Configuration menu
    Copy the full SHA
    e2b1a78 View commit details
    Browse the repository at this point in the history