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. |
||
|---|---|---|
| .. | ||
| e2e_testing | ||
| examples | ||
| python | ||
| test | ||
| tools | ||
| CMakeLists.txt | ||