search

iree-org / iree

A retargetable MLIR-based machine learning compiler and runtime toolkit.
http://iree.dev/
Apache License 2.0
2.47k stars 548 forks source link

[LinalgExt] Add im2col tiling implementation #17671

Closed Max191 closed 3 hours ago

Max191 commented 1 week ago

This adds the tiling interface implementation for iree_linalg_ext.im2col. The implementation is simple, since the M and K dimensions of the input to not get tiled for the im2col op. The output is fully tiled. and only the batch dimension is tiled for the input. The main job of the tiling implementation is to tile the result tensor and set the correct offsets into the input tensor for the op.

Max191 commented 1 week ago

This is based on https://github.com/iree-org/iree/pull/17644, please only review the last commit.