transformations: Decouple distribute-stencil from shape-inference

[PapyChacal]

Make dmp shape inference part of the general shape-inference, and remove shape-inference as a part of distribute-stencil.

Stacked on:

• dialects: dmp: make strategies attributes and carry them in IR #3050

[codecov]

Codecov Report

All modified and coverable lines are covered by tests ✅

Project coverage is 89.88%. Comparing base (f9df366) to head (ff58a95).
Report is 1 commits behind head on main.

Additional details and impacted files

@@           Coverage Diff           @@
##             main    #3054   +/-   ##
=======================================
  Coverage   89.88%   89.88%           
=======================================
  Files         415      416    +1     
  Lines       52426    52436   +10     
  Branches     8111     8114    +3     
=======================================
+ Hits        47121    47131   +10     
  Misses       4000     4000           
  Partials     1305     1305           

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

[n-io]

Looks good, thanks! Please can you leave a comment whether dmp needs to run before/after shape inference, or does either work equally well?

[PapyChacal]

Looks good, thanks! Please can you leave a comment whether dmp needs to run before/after shape inference, or does either work equally well?

Shape inference would need to run after dmp. DMP is just structurally placing "when" communications might need to happen. Shape inference allows to infer what needs to be communicated at those points.
Ideally, shape inference could happen before too, and just happen again on the swaps if needed. This is not tested at this point though, and I would expect it to fail from experience.

This drop-in replacement order is indicated by the update of distribute-stencil.mlir