b636e0c40c
[Stablehlo Dialect] fix lowering batch_norm with mixed types ( #2383 )
...
* [Stablehlo Dialect] fix lowering bn inference with mixed types
* update
2023-08-21 17:36:56 +08:00
41bafe13cc
[build] Update llvm tag to a3f2751f ( #2397 )
...
This commit updates the `llvm-project` and `mlir-hlo` submodules to
commits:
llvm-project: a3f2751f782f3cdc6ba4790488ec20163a40ac37
mlir-hlo: 97c7e4b4506c3a2441c923e592833f45da439009
Changes made:
- Rename `getSuccessorEntryOperands` with `getEntrySuccessorOperands`
and remove `operands` from
`getSuccessorRegions` (https://reviews.llvm.org/D157506 )
- Make `TypeConverter` a `const` (https://reviews.llvm.org/D157601 )
2023-08-15 09:53:28 -07:00
60bad54f27
[Torch Dialect] replace none-index in aten.Index.Tensor's param by manually generating it ( #2344 )
...
* [Torch Dialect] replace none-index in aten.Index.Tensor's param by manually generating it
Co-authored-by: Jiawei Wu <wujiawei.aml@bytedance.com>
Co-authored-by: Jianzhe Xiao <jianzhe.xiao@bytedance.com>
* minor typo fix
* add new failed e2e tests for ltc
* fix typo
* Address comments
* Add more e2e tests
* add failed e2e tests for LTC
* address comments
* remove decomposition for AtenIndexTensorHackedTwinOp
2023-08-15 19:36:08 +08:00
2fbb4c79f0
Fix "unused variable 'outType'" warning ( #2378 )
2023-08-05 07:41:01 +08:00
16923fdbd2
[Stablehlo] Add converter to stablehlo for aten.(Int,Float,Bool).Tensor op ( #2340 )
...
[Stablehlo] Add converter to stablehlo for aten.(Int,Float,Bool).Tensor op and configure crashing e2e sets for stablehlo backend.
2023-07-29 21:55:49 +08:00
c7c59b540e
[Stablehlo] support dynamic shape when convert aten.fill.Scalar ( #2349 )
2023-07-27 18:35:25 +08:00
31ef08b63d
[Stablehlo]Add support for AvgPool1dOp ( #2268 )
...
* Add support for AvgPool1d
* Update AbstractInterpLibrary
* support avgpool1d in linalg
* refactored code
* fix nit problem
2023-07-25 14:09:53 +08:00
026e8db2e4
[Stablehlo] add converter for aten.scatter.src op ( #2295 )
2023-07-24 10:14:45 +08:00
9535be7903
[Torch-Dialect] emit aten.narrow.Tensor op and decompose it to aten.narrow op ( #2297 )
2023-07-20 16:46:44 +08:00
6a072d4f4a
[Stablehlo] AtenEmptyMemoryFormat remove device cpu check ( #2288 )
...
* remove cpu check
* update dtype
---------
Co-authored-by: zhekun.zhang <zhekun.zhang@bytedance.com>
2023-07-10 15:36:21 +08:00
0548e2ef3b
[Stablehlo] fix promoteType() when input doesn't have DefiningOp ( #2262 )
2023-06-26 00:04:17 +08:00
bba0f5891b
[Stablehlo] add conversion for AtenFlipOp ( #2163 )
2023-06-15 10:27:34 +08:00
0caaf8d32a
Bump LLVM ( #2176 )
...
* Bump LLVM
---------
Co-authored-by: Matthias Gehre <matthias.gehre@xilinx.com>
2023-06-13 16:17:23 +02:00
f0b7b63be0
[Stablehlo] Add aten.uniform lowering ( #2101 )
...
* add uniform stablehlo lowering
* add unit test
* new line
* rm redundant file
* Empty commit, trigger test
* fix include
* address comments
---------
Co-authored-by: zhekun.zhang <zhekun.zhang@bytedance.com>
2023-05-25 10:32:55 +08:00
eb8f56aeb7
[Stablehlo] Add `AtenIndexTensor` StableHlo support ( #2107 )
...
* Add AtenIndexTensor StableHlo support
* clean up
* Empty commit, trigger test
* try to debug hanging test
* fix segfulat
* fix bad include
---------
Co-authored-by: zhekun.zhang <zhekun.zhang@bytedance.com>
2023-05-24 11:13:57 -07:00
1333674905
[StableHlo] Support AtenEmptyMemoryFormatOp ( #2092 )
...
* add empty conversion
* clean up
* add tests
---------
Co-authored-by: zhekun.zhang <zhekun.zhang@bytedance.com>
2023-05-18 19:07:35 -07:00
6f7d9e83df
[Stablehlo] add e2e test for aten.batch_norm ( #2129 )
2023-05-17 09:04:40 -07:00
1eb18dd8b5
Add AtenFillScalarOp Stablehlo support ( #2108 )
...
Co-authored-by: zhekun.zhang <zhekun.zhang@bytedance.com>
2023-05-11 16:41:46 -07:00
51e0a2c933
[Stablehlo] Add stablehlo support for aten.abs ( #2068 )
...
Co-authored-by: AmosLewis <Amos_Lewsi@foxmail.com>
2023-05-08 22:13:00 -07:00
0096ceae2f
[Stablehlo] fix reduce max init_value with -inf ( #2064 )
...
* [Stablehlo] fix reduce max init_value with -inf
* update
2023-05-06 12:05:51 -07:00
fc62b8e9ab
[StableHlo] Fix AtenWhereSelfOp convert rule ( #2093 )
...
* fix whereself convert rule
* use int to test promotion
* add dynamo failing test
---------
Co-authored-by: zhekun.zhang <zhekun.zhang@bytedance.com>
2023-05-05 15:21:55 -07:00
1eceb84899
add stablehlo support for pow.tensor_tensor ( #2086 )
...
Co-authored-by: zhekun.zhang <zhekun.zhang@bytedance.com>
2023-05-04 09:55:03 -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
72bb902640
[STABLEHLO] Move utils.h to include/ ( #1974 )
2023-03-27 21:16:21 -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
3698a95586
[MHLO] add conversion for aten.linalg_vector_norm ( #1850 )
2023-03-20 14:14:27 -07:00
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
089018b658
[MHLO] move AtenTanhOp to ConvertAtenUnaryFPOnlyPatten and add sin/cos/ceil/floor pattern ( #1847 )
2023-02-06 11:14:26 -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
Yuanqiang Liu
Ramiro Leal-Cavazos
Jiawei Wu
JianzheXiao
Zhekun Zhang
Maksim Levental
Chi_Liu
Abhishek Varma
Ziheng Jiang
Ashay Rane