mirror of https://github.com/llvm/torch-mlir
00efec0b73
* Enables assume_strict_symbolic_shapes on fx_importer imported programs, indicating strict shape semantics. * Reworks the view->reshape lowering to take advantage of strict mode and do one of: * Collapse to 0D * Flatten/Unflatten when there is an inferred dim. * Fallback to tensor.reshape * Splits some test cases up and adds an attribute to control the old pattern (so new corners can be tested in strict mode in isolation). * Dynamic inferred mode needs upstream work to generalize expand_shape (so that case is suppressed here). * Deletes the assert from the existing tensor.reshape lowering if strict shape mode is enabled (since the condition it is dynamically asserting cannot happen). |
||
|---|---|---|
| .. | ||
| TorchConversionToMLProgram | ||
| TorchOnnxToTorch | ||
| TorchToArith | ||
| TorchToLinalg | ||
| TorchToSCF | ||
| TorchToStablehlo | ||
| TorchToTMTensor | ||
| TorchToTensor | ||
| TorchToTosa | ||
| Utils | ||
| CMakeLists.txt | ||
| PassDetail.h | ||
| Passes.cpp | ||