torch-mlir/lib/Dialect/Torch
Ramiro Leal-Cavazos ea371a9bf2
Fix handling of view-like ops in `maximize-value-semantics` (#611)
This commit adds handling to the `maximize-value-semantics` pass for
the case where a view-like op depends on a tensor that has been
overwritten by a value tensor. The approach for removing the
dependency is to change the input to the view-like op to be a copy of
the value tensor that is being used to overwrite.

This commit also removes `AtenFill_ScalarOp` and
`AtenBernoulli_FloatOp` from the list of view-like ops, since these
ops now have a corresponding op with value semantics into which they
get converted in the `reduce-op-variants` pass.
2022-02-18 10:19:07 -08:00
..
IR LLVM bump 2022-02-16 15:28:13 -05:00
Transforms Fix handling of view-like ops in `maximize-value-semantics` (#611) 2022-02-18 10:19:07 -08:00
Utils [LINALG] Decompose `aten.batch_norm` into `aten.native_batch_norm` 2022-02-16 23:41:38 +05:30
CMakeLists.txt E2e support for aten.softmax.int and aten.embedding 2021-10-18 17:57:45 -04:00