HomePhabricator

[mlir] Use affine.apply when distributing to processors

Authored by antiagainst on Mar 9 2021, 5:36 AM.

Description

[mlir] Use affine.apply when distributing to processors

This makes it easy to compose the distribution computation with
other affine computations.

Reviewed By: mravishankar

Differential Revision: https://reviews.llvm.org/D98171