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). |
||
|---|---|---|
| .. | ||
| CMakeLists.txt | ||
| DataMovement.cpp | ||
| IndirectDataMovement.cpp | ||
| Linear.cpp | ||
| Pooling.cpp | ||
| PopulatePatterns.h | ||
| Random.cpp | ||
| Reduction.cpp | ||
| TensorConstructors.cpp | ||
| TensorScalarInterop.cpp | ||
| TorchToLinalg.cpp | ||
| Uncategorized.cpp | ||
| Utils.cpp | ||