4d98f76d4f
[Torch Dialect] fold aten.detach ( #2021 )
2023-04-18 08:59:14 -07:00
b31c5fa1fb
build: update llvm tag to 225d255a
...
Summary of changes:
- Update tags
llvm: 225d255a583ea3d50bbba49d949ca76be6a880bf
mhlo: beb5d2b04c33c19d23e02bd3c5e50ddecba11664
Fixes https://github.com/llvm/torch-mlir/issues/1859 .
Signed-Off By: Vivek Khandelwal <vivek@nod-labs.com>
2023-04-15 06:27:10 +05:30
f3d1eda09f
[TOSA] Add aten.abs support ( #2032 )
2023-04-14 08:43:39 -07:00
ad36e61040
[TOSA] Add aten.le.tensor support ( #2033 )
2023-04-14 08:43:14 -07:00
318fe13468
[MLIR][TORCH] Patch up Ops and their lowerings to deal with +ve `dim`
...
-- In Python we have the concept of negative dimension indexing.
-- We would want to normalize such dimensions to be +ve and within the
expected range instead.
-- This commit takes care of a few remaining set of Ops and their
lowerings by applying `toPositiveDim` and `isValidDim` to the
extracted integer `dim` value.
Signed-off-by: Abhishek Varma <abhishek@nod-labs.com>
2023-04-14 13:12:56 +05:30
1bd5747ca3
[StableHlo] Fix transposed convolution conversion ( #2026 )
...
* fix conv bwd
* fix
* fix group case
* clean up
---------
Co-authored-by: zhekun.zhang <zhekun.zhang@bytedance.com>
2023-04-13 11:24:39 -07:00
c2c96c430a
Add Shape inference for CopyOp for lazy tensor core backend ( #2006 )
...
- Add Shape inference for CopyOp for LTC backend
2023-04-12 09:37:03 -04:00
72c3326097
[Torch Dialect] support for aten.one_hot ( #1852 )
2023-04-11 01:02:28 -07:00
98747d09a8
[MLIR][TORCH] Add support for prims::view_of op
...
This op does nothing and just returns the input operand as the
result of the op.
Signed-Off By: Vivek Khandelwal <vivek@nod-labs.com>
2023-04-11 07:58:10 +05:30
2213ce0855
[TorchDynamo] Add aten.squeeze op to the decomposition list
...
Signed-Off By: Vivek Khandelwal <vivek@nod-labs.com>
2023-04-06 22:21:25 +05:30
e90ea3d7ab
[MLIR][TORCH] Extend implementation of aten._index_put_impl op.
...
This commits adds the support for cases for index_put_op:
1.) where index is a 2-d tensor.
2.) where indices is a list of tensors and none, with exactly
2 non none tensors along the consecutive dimensions.
This commit also adds a utility to compute the broadcast shape
given the two input tensors.
Signed-Off By: Vivek Khandelwal <vivek@nod-labs.com>
2023-04-05 14:04:30 +05:30
788efc3180
[MLIR][TORCH] Add support for non-unit stride for conv backward
...
This commit also adds the support for non-unit output padding in the
case of transposed convolution.
Signed-Off By: Vivek Khandelwal<vivek@nod-labs.com>
2023-04-04 17:53:27 +05:30
5e9582b055
[MLIR][TORCH] Add e2e support aten.movedim.int op
...
Signed-Off By: Vivek Khandelwal<vivek@nod-labs.com>
2023-04-04 17:53:27 +05:30
82fb9c7fb8
[MLIR][TORCH] Add decomposition for prims::squeeze op
...
This commit adds the decomposition for the prims.squeeze op.
Signed-Off By: Vivek Khandelwal<vivek@nod-labs.com>
2023-04-01 21:45:58 +05:30
6bb9965a41
[TOSA] Add support for AtenZerosOp 0/strided layout ( #1983 )
2023-03-30 07:08:20 -07:00
2389729fb9
Add support for aten_remainder in TorchToTosa ( #1966 )
2023-03-23 17:55:58 -07:00
5758a0bfbb
[StableHLO] Support for slice_scatter ( #1960 )
...
Co-authored-by: zhekun.zhang <zhekun.zhang@bytedance.com>
2023-03-22 13:41:04 -07:00
d632afce31
Max pool2d ceil mode to tosa ( #1957 )
...
* implemented ceil_mode== true support for lowering aten.max_pool2d to tosa
* add e2e test for lowering aten.max_pool2d to tosa with ceil_mode=true
---------
Co-authored-by: Lisa Liu <lingl@xilinx.com>
2023-03-21 10:17:39 -07:00
3698a95586
[MHLO] add conversion for aten.linalg_vector_norm ( #1850 )
2023-03-20 14:14:27 -07:00
7d711b9f9f
Constant pad nd to tosa ( #1933 )
...
* implemented lowering torch.aten.constant_pad_nd to tosa
* add constant_pad_nd e2e tests to TOSA_PASS_SET
* add PadModule_basic & PadWithNoneValModule_basic to TOSA_PASS_SET
---------
Co-authored-by: Lisa Liu <lingl@xilinx.com>
2023-03-15 08:42:15 -07:00
4912c3937d
Support aten.stack op and decompose it into unsqueeze & cat ( #1747 )
2023-03-11 09:25:25 +08:00
66b1045a80
Add a new RecomposeComplexOps pass, fold slice+copy_ into indeX_put_ ( #1901 )
2023-03-10 16:42:11 -05:00
dca2b8a40a
[TORCH] Improve type refinement for aten.cat. ( #1908 )
...
* [TORCH] Fix type refinement for aten.cat.
* Add test.
* Address comments.
* Update.
* Update.
* Update.
* Update.
* Update.
---------
Co-authored-by: Ziheng Jiang <ziheng.jiang@bytedance.com>
2023-03-09 16:17:35 -08:00
1d3a7419c5
[Torch Dialect] add RSub, ScalarImplicit canonicalize ( #1899 )
...
* add rsub, scalarimplit canonicalizer
* reformat
* address comments
* fix bug
* fix test
* Update elementwise.py
* resolve merge conflict
* change to 3
* change to 3
* real fix
* fix name
* add torchdynamo fail test
---------
Co-authored-by: zhekun.zhang <zhekun.zhang@bytedance.com>
2023-03-06 17:38:27 -08:00
c2ef5f4165
Add HardtanhBackward TOSA and LINALG support ( #1721 )
2023-03-06 10:16:37 -08:00
671be048fe
Fix handling of non-int tensors in `getScalarValue` ( #1914 )
...
The current implementation of `getScalarValue` does not check that the
input to a `ValueTensorLiteralOp` is an i64 before extracting the
value, and it does not check that the result type of the
`PrimNumToTensorScalarOp` is also an i64. This leads to crashes or
invalid IR generated when the `input` is something other than an i64
tensor or `!torch.int`.
This commit addresses those issues. In addition, the function
`getScalarValue` is renamed to `getScalarIntValue` to make it clear
that it *only* extracts scalar integers.
2023-03-06 10:12:58 -08:00
7a8304f935
[Torch Dialect] add folder for aten.sub.float ( #1871 )
2023-03-02 09:07:33 -08:00
a32840ffd7
build: manually update PyTorch version
...
Set PyTorch and TorchVision version to nightly release 2023-02-27.
This commit also adds the lowering for aten.add and aten.Float.Scalar op.
Signed-Off By: Vivek Khandelwal <vivek@nod-labs.com>
2023-02-28 22:43:39 +05:30
0afb85d45f
build: update llvm tag to 5e111eb275
...
- Update llvm tag to 5e111eb275eee3bec1123b4b85606328017e5ee5
- mhlo now points to a99159c45ee5c497f8dce01eff807a6d57629b61
Signed-Off-by: Gaurav Shukla <gaurav@nod-labs.com>
2023-02-23 22:57:17 +05:30
6a3438f672
build: manually update PyTorch version
...
Set PyTorch and TorchVision version to nightly release 2023-02-20.
Signed-Off By: Vivek Khandelwal <vivek@nod-labs.com>
2023-02-23 11:33:41 +05:30
38ed559398
[StableHLO] Add support for AtenPowTensorScalar. ( #1883 )
...
* [MHLO] Add support for AtenPowTensorScalar.
* Update.
---------
Co-authored-by: Ziheng Jiang <ziheng.jiang@bytedance.com>
2023-02-16 20:26:46 -08:00
b17d4d4f08
[MLIR][TORCH] Add decomposition for aten.bernoulli.p op ( #1882 )
...
Signed-Off By: Vivek Khandelwal <vivek@nod-labs.com>
2023-02-15 22:36:29 +05:30
f6f2e4d040
[MLIR][TORCH] Add support for integer type input for max.dim op
...
Signed-Off By: Vivek Khandelwal <vivek@nod-labs.com>
2023-02-15 16:14:15 +05:30
8a7340dfb5
[TOSA] aten.index.tensor multiple indexes support ( #1868 )
2023-02-13 23:07:15 -08:00
f1b8d5e581
[MHLO] Support AtenMaskedFillScalar ( #1839 )
...
* [MHLO] Support MaskedFillScalar.
* Update.
* Update.
* Update.
---------
Co-authored-by: Ziheng Jiang <ziheng.jiang@bytedance.com>
2023-02-10 13:58:39 -08:00
cc819e73dd
[TOSA] Fix broadcast_to input and output different shape support ( #1855 )
2023-02-09 09:15:14 -08:00
282c4e027b
build: manually update PyTorch version
...
Set PyTorch and TorchVision version to nightly release 2023-02-07.
Fixes https://github.com/llvm/torch-mlir/issues/1792 .
Signed-Off By: Vivek Khandelwal <vivek@nod-labs.com>
2023-02-08 11:41:58 +05:30
3ebe5a5a67
build: update llvm tag to ba8b8a73f ( #1856 )
...
* build: update llvm tag to ba8b8a73f
* skip ConvolutionModule2DGroups_basic
2023-02-07 23:25:59 +08:00
3e60e6021f
build: manually update PyTorch version
...
Set PyTorch and TorchVision version to nightly release 2023-02-06.
Signed-Off By: Vivek Khandelwal <vivek@nod-labs.com>
2023-02-07 12:02:18 +05:30
089018b658
[MHLO] move AtenTanhOp to ConvertAtenUnaryFPOnlyPatten and add sin/cos/ceil/floor pattern ( #1847 )
2023-02-06 11:14:26 -08:00
c957cebd03
build: manually update PyTorch version
...
Set PyTorch and TorchVision version to nightly release 2023-02-05.
Signed-Off By: Vivek Khandelwal <vivek@nod-labs.com>
2023-02-06 13:23:28 +05:30
f58ba19448
Add aten.bucketize op and its decomposition ( #1834 )
2023-02-03 10:20:47 +08:00
711646d095
mhlo: migrate conversion to stablehlo ( #1840 )
...
This patch replaces all MHLO operations with their StableHLO
counterparts and adds a validation pass to ensure that no MHLO operations
remain before translating all Stablehlo operations to the MHLO dialect
for further lowering to the Linalg dialect.
This patch also updates all lit tests so that they refer to the
`convert-torch-to-stablehlo` pass and so that they check for StableHLO
operations.
2023-02-02 07:29:47 -06:00
ed9d8d1fb7
[MLIR][TORCH] Add support for clone op with channels last memory format
...
Fixes https://github.com/llvm/torch-mlir/issues/1829
Signed-Off By: Vivek Khandelwal <vivek@nod-labs.com>
2023-02-02 16:04:42 +05:30
f5b689e12f
[MHLO] Support aten.cumsum op in mhlo backend ( #1825 )
2023-01-29 21:38:27 -08:00
c5ac42a198
[TOSA] Add aten.view shape -1 support ( #1815 )
2023-01-20 11:56:26 -08:00
2587b3f583
[TOSA] Add aten.Index.Tensor support ( #1771 )
2023-01-19 21:19:00 -08:00
abf4f207cd
[MLIR][TORCH] Add canonicalizer for aten.new_empty_strided op
...
Signed-Off By: Vivek Khandelwal<vivek@nod-labs.com>
2023-01-19 13:37:32 +05:30
f9d59eb500
[MLIR][TORCH] Add decomposition for aten.randn_like op
...
This commit decomposes aten.randn_like op into aten.randn.generator op.
Signed-Off By: Vivek Khandelwal <vivek@nod-labs.com>
2023-01-18 12:09:27 +05:30
e2698433db
Fix empty tensor when select -1 ( #1787 )
2023-01-17 10:14:14 -08:00
Yuanqiang Liu
Vivek Khandelwal
Chi_Liu
Abhishek Varma
Zhekun Zhang
rahuls-cerebras
Michael Feliz
lisaliu1
Jiahao Li
gpetters94
Ziheng Jiang
Priya Savithiri
Ramiro Leal-Cavazos
Gaurav Shukla
Tanyo Kwok
Ashay Rane