[tilingInterface] Update the tile sizes to i64 attr type

iree-org / iree

A retargetable MLIR-based machine learning compiler and runtime toolkit.

http://iree.dev/

Apache License 2.0

2.47k stars 551 forks source link

[tilingInterface] Update the tile sizes to i64 attr type #17761

Closed pashu123 closed 2 days ago

pashu123 commented 3 days ago

Update the tile sizes to contain i64Attrs instead of arith.constant. Somehow it's giving dynamic shapes in tensor.extract_slice since the arith.constant op isn't folded or seen as a constant.

To fix Issue: https://github.com/iree-org/iree/issues/17441