mirror of https://github.com/llvm/torch-mlir
3dc7847348
Some of the lowerings use the result type obtained from the op itself to tell the `linalg::GenericOp` what the type of the result should be rather than using the type of the result tensor given to the `linalg::GenericOp`. This becomes a problem when the result type of the op has static size information and the result tensor used in `linalg::GenericOp` has dynamic dimensions, for `linalg::GenericOp` expects the result type to be equal to the type of the output tensor. This commit replaces the use of the result type from the op itself with the type of the result tensor passed to `linalg::GenericOp`. In order to not create too many dynamic/static versions of the same e2e test, e2e tests have only been added to the ops that currently fail when used with static sizes. |
||
|---|---|---|
| .. | ||
| TorchToLinalg | ||
| TorchToSCF | ||
| TorchToStd | ||
| TorchToTosa | ||
| CMakeLists.txt | ||
| PassDetail.h | ||
| Passes.cpp | ||