spcl / daceml

A Data-Centric Compiler for Machine Learning
https://daceml.readthedocs.io
BSD 3-Clause "New" or "Revised" License
81 stars 14 forks source link

Support nested SDFGs in distributed lowering #123

Open orausch opened 2 years ago

orausch commented 2 years ago

This allows us to support reductions with their intialization states. The idea is that nested SDFG are required to be schedule such that there is no communication within them. The user passes the schedules for each map, and the implied communication constraints are then checked for consistency.

Keeping communication out of the Nested SDFGs means that there is no communication between things like reduction buffer initialization, and also means that all global communication is kept top-level, where it is easier to optimize

codecov[bot] commented 2 years ago

Codecov Report

:exclamation: No coverage uploaded for pull request base (spr/orausch/master.support-nested-sdfgs-in-distributed-lowering@00d1787). Click here to learn what that means. The diff coverage is n/a.

@@                                        Coverage Diff                                         @@
##             spr/orausch/master.support-nested-sdfgs-in-distributed-lowering     #123   +/-   ##
==================================================================================================
  Coverage                                                                   ?   10.47%           
==================================================================================================
  Files                                                                      ?       68           
  Lines                                                                      ?     7564           
  Branches                                                                   ?        0           
==================================================================================================
  Hits                                                                       ?      792           
  Misses                                                                     ?     6772           
  Partials                                                                   ?        0           

:mega: We’re building smart automated test selection to slash your CI/CD build times. Learn more