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
924503c436
build: manually update PyTorch version
...
Set PyTorch and TorchVision version to nightly release 2023-01-15.
Signed-Off By: Vivek Khandelwal <vivek@nod-labs.com>
2023-01-16 13:07:57 +05:30
4f94831fed
[LINALG][TOSA][MHLO] Add e2e support for aten bitwise ops ( #1753 )
2023-01-11 14:40:03 -08:00
fd236b2c89
[MLIR][TORCH] Add decomposition for prims.var and prims.sqrt op
...
Signed-Off By: Vivek Khandelwal <vivek@nod-labs.com>
2023-01-11 17:39:10 +05:30
4e4a571104
[TOSA] Add LeakyReLU conversion pass ( #1790 )
...
* feat(TorchToTOSA): LeakyReLU legalization
* test(LeakyReLU): Add LIT test and enable e2e test
Co-authored-by: Philipp Braun <philipp.braun@amd.com>
2023-01-10 21:42:07 -08:00
8f01072099
Fix OptionalCType class name ( #1779 )
...
* Fix OptionalCType class name
* Rmove LTC xfail tests
2023-01-06 17:03:24 -05:00
8dc5d985eb
Add e2e support for aten logical or/and/xor/not ops ( #1761 )
2023-01-03 18:11:25 -08:00
a88e3766e8
Add E2E support for LeakyRelu and LeakyReluBackward ops ( #1733 )
...
Co-authored-by: srirammaswamy <srirammaswamy@gmail.com>
2023-01-03 08:30:16 -08:00
b2dbbba5e6
xfail ElementwisePreluModule_basic in dynamo ( #1763 )
2022-12-31 13:54:25 -08:00
ac780529b4
Revert e2e support for aten logical or/and/xor/not ops ( #1757 )
...
This reverts commit eaab9be207
2022-12-29 21:01:06 -06:00
2f45959f0d
Prelu lowering to linalg ( #1712 )
...
Prelu lowering to linalg
2022-12-28 08:51:33 +05:30
eaab9be207
Add e2e support for aten logical or/and/xor/not ops ( #1752 )
2022-12-26 10:23:38 +08:00
49071f86e6
[MHLO] Evaluate RuntimeAssertOp at compile time ( #1732 )
2022-12-22 17:12:52 +08:00
60a139271d
Add aten.std.correction op and its decomposition ( #1731 )
2022-12-21 21:02:40 -08:00
Vivek Khandelwal
Chi_Liu
Michael Feliz
Zhekun Zhang
lisaliu1
Yuanqiang Liu
Jiahao Li
gpetters94
Ziheng Jiang
Priya Savithiri
Ramiro Leal-Cavazos
Gaurav Shukla
Tanyo Kwok
Ashay Rane
Gleb Kazantaev
Srirammaswamy
powderluv
Shivam Gupta