Tags: xndcn/torch-mlir
Tags
Lower aten::view with linalg.collapse and linalg.expand We only handle the expanding OR collapsing cases, we do not handle expanding And collapsing happening at the same time or cases where it's neither collapsing nor expanding like view of [2,3] for 3x2 tensor. It's assumed that if a shape list element is got from `aten.size(tensor, dim)` the corresponding dim is not splitted or collapsed. This assumption makes it easier to deal with dynamic shapes.
[TORCH][MLIR] Add E2E support for `aten.empty_like` op This commit adds decomposition of `aten.empty_like` into `aten.empty` op. Signed-Off-by: Gaurav Shukla <gaurav@nod-labs.com>
* [tosa] Support for Rsqrt legalization (llvm#480) Signed-off-by: Anup Gangwar <anup.gangwar@arm.com> Co-authored-by: Anup Gangwar <anup.gangwar@arm.com>
* [tosa] Support for Rsqrt legalization (llvm#480) Signed-off-by: Anup Gangwar <anup.gangwar@arm.com> Co-authored-by: Anup Gangwar <anup.gangwar@arm.com>
Add support for int types in gtScalar op. Support for integer types in gtScalar op has been added. The code share same logic with gtTensor op and can be merged which is added as a TODO.
Add support for int types in gtScalar op. Support for integer types in gtScalar op has been added. The code share same logic with gtTensor op and can be merged which is added as a TODO.
Add `aten.gt.Tensor` op `aten.gt.Tensor` op has been added in torch dialect and the lowering of the op has been done to the linalg dialect. Signed-off-by: Prashant Kumar <prashant@nod-labs.com>
Add `aten.gt.Tensor` op `aten.gt.Tensor` op has been added in torch dialect and the lowering of the op has been done to the linalg dialect. Signed-off-by: Prashant Kumar <prashant@nod-labs.com>
PreviousNext