iree icon indicating copy to clipboard operation
iree copied to clipboard
verified publisher icon iree-org

[EPIC][CPU] Enable direct pack/unpack vectorization in IREE

Open hanhanW opened this issue 1 year ago • 1 comments

This collects the tasks to enable direct pack/unpack vectorization by default.

Tasks

  • [x] Land https://github.com/iree-org/iree/pull/16692
  • [x] Finish vector input sizes inference from IR for pack op. https://github.com/openxla/iree/pull/16629
  • [x] Finish vector input sizes inference from IR for unpack op. https://github.com/openxla/iree/pull/16664
  • [ ] https://github.com/openxla/iree/issues/16975
  • [ ] https://github.com/openxla/iree/issues/16979
  • [x] Deprecate DecomposePackUnPackOps pass. I.e., land https://github.com/openxla/iree/pull/16672/commits/e7641c75209605debd8fe69914e3a9fc74c44b9b

hanhanW avatar Apr 02 '24 20:04 hanhanW

The two prototypes I have for pack/unpack shape inference are https://github.com/openxla/iree/pull/16629 and https://github.com/openxla/iree/pull/16664

hanhanW avatar Apr 03 '24 19:04 hanhanW