torch-mlir/python
Sambhav Jain 0b2f9c89a2
Bring back `dynamic_shapes` constraints in fx importer API (#3026)
https://github.com/llvm/torch-mlir/pull/2992 dropped `constraints` from
the fx importer API,
[breaking](https://github.com/cruise-automation/mlir-tcp/actions/runs/8284385380/job/22669774071)
downstream AOT compile tests in `mlir-tcp` that use it. This knob has
been soft-deprecated for a while now, replaced by `dynamic_shapes` - a
more ergonomic interface. This PR brings back dynamic_shapes constraints
in the new supported form. Also added a python lit test with dynamic
shaped annotations.
2024-03-14 10:26:34 -07:00
..
torch_mlir Bring back `dynamic_shapes` constraints in fx importer API (#3026) 2024-03-14 10:26:34 -07:00
CMakeLists.txt add support for decomposition (#2879) 2024-02-14 21:00:52 -08:00
TorchMLIRModule.cpp Upstream the ONNX importer. (#2636) 2023-12-12 19:02:51 -08:00