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). |
||
|---|---|---|
| .. | ||
| basic.mlir | ||
| broadcast.mlir | ||
| convolution.mlir | ||
| elementwise.mlir | ||
| flatten.mlir | ||
| gridsampler.mlir | ||
| pooling.mlir | ||
| resize.mlir | ||
| sparse.mlir | ||
| unsqueeze.mlir | ||
| view.mlir | ||
| view_strict.mlir | ||