torch-mlir/lib/Dialect/Torch
Sean Silva 776426ea4e [SimplifyShapeCalculations] Fix AbstractlyInterpretListOpsWithinABlock
The logic in the rewriting phase had a bug in case of a read-only op
coming before mutation ops. The logic would use the op itself as the
"latest literal", but that is not correct, because later on we replace
the op itself with the *final* "latest literal", assuming that all uses
of the op have been rewritten -- that was working in general, except for
any read-only ops at the beginning.

Big thanks to @ljfitz for the tiny reproducer!

Fixes #704
2022-03-28 13:18:35 -07:00
..
IR Add non-default dtype support for a few elementwise math ops. (#687) 2022-03-23 13:35:43 -07:00
Transforms [SimplifyShapeCalculations] Fix AbstractlyInterpretListOpsWithinABlock 2022-03-28 13:18:35 -07:00
Utils Avoid `using` the `torch_upstream` namespace. 2022-03-15 17:24:17 -07:00
CMakeLists.txt E2e support for aten.softmax.int and aten.embedding 2021-10-18 17:57:45 -04:00