mirror of https://github.com/llvm/torch-mlir
03e8f99730
Adds support for lowering to prims split_op. Similar design to collapse op lowering in https://github.com/llvm/torch-mlir/pull/2572, with some small differences, because the split_dim op (in pytorch) is view-changing whereas the collapse is not. The difference means that 1) it must be registered in the function Torch::isViewLikeOp 2) it must be be added to the "expected fail" set for the torch dynamo backend. |
||
|---|---|---|
| .. | ||
| CMakeLists.txt | ||
| TorchUpstream.cpp | ||
| Utils.cpp | ||