search

NVlabs / timeloop

Timeloop performs modeling, mapping and code-generation for tensor algebra workloads on various accelerator architectures.
https://timeloop.csail.mit.edu/
BSD 3-Clause "New" or "Revised" License
303 stars 99 forks source link

[isl-spatial-analysis] Add unicast hops calculation #246

Closed gilbertmike closed 3 months ago

gilbertmike commented 4 months ago

This PR adds unicast hops calculation into the ISL nest analysis path. Specifically, the PR adds:

  1. An analysis inside ISL multicast analysis that computes unicast hops.
  2. A member in the ISL analysis output data structure to hold the unicast hops.
  3. Update to the ISL analysis output to nest analysis data structure adaptor to pass along the unicast hops value.
  4. Update to the unit test for ISL multicast analysis to test the new analysis.