forked from OSchip/llvm-project
36a415bcc5
Add one more simplification for floordiv and mod affine expressions.
Examples:
(2*d0 + 1) floordiv 2 is simplified to d0
(8*d0 + 4*d1 + d2) floordiv 4 simplified to 4*d0 + d1 + d2 floordiv 4.
etc.
Similarly, (4*d1 + 1) mod 2 is simplified to 1,
(2*d0 + 8*d1) mod 8 simplified to 2*d0 mod 8.
Change getLargestKnownDivisor to return int64_t to be consistent and
to avoid casting at call sites (since the return value is used in expressions
of int64_t/index type).
Signed-off-by: Uday Bondhugula <uday@polymagelabs.com>
Closes tensorflow/mlir#202
COPYBARA_INTEGRATE_REVIEW=https://github.com/tensorflow/mlir/pull/202 from bondhugula:affine b13fcb2f1c00a39ca5434613a02408e085a80e77
PiperOrigin-RevId: 284866710
|
||
|---|---|---|
| .. | ||
| compose_maps.mlir | ||
| normalize_maps.mlir | ||
| vector_utils.mlir | ||
| vectorize_1d.mlir | ||
| vectorize_2d.mlir | ||
| vectorize_3d.mlir | ||
| vectorize_outer_loop_2d.mlir | ||
| vectorize_outer_loop_transpose_2d.mlir | ||
| vectorize_transpose_2d.mlir | ||