torch-mlir/lib/Dialect/Torch
Ashay Rane 29bc48aedb
torch: add pass to catch non-value tensors (#1052)
This patch adds a new pass `torch-verify-conversion-to-value-semantics`,
which looks for non-value semantics tensors to catch such tensors early
during compilation.

This pass requires `torch-refine-public-return` pass to ensure that
return operations are updated to use value tensors, followed by the
canonicalize pass to remove any dead ops that may use or produce
non-value tensors.
2022-07-13 17:11:15 -07:00
..
IR canonicalizer: propagate type information across copy and cast ops (#1030) 2022-07-12 12:38:37 -07:00
Transforms torch: add pass to catch non-value tensors (#1052) 2022-07-13 17:11:15 -07:00
Utils Fix type promotion code for scalar only operations 2022-05-07 10:37:21 -04:00
CMakeLists.txt E2e support for aten.softmax.int and aten.embedding 2021-10-18 17:57:45 -04:00