51ef1b141c
Add some missing dependencies.
...
Caught in the wild here:
https://github.com/llvm/torch-mlir/runs/8046660640?check_suite_focus=true
It is common for a missing dependency to only surface as an issue on the
CI machines since they have fewer cores which prevents a "race" that
happens to cause the dependency to be built before the dependent.
2022-08-30 11:52:30 -07:00
1d9d925f6e
mlir: fix replacement of `OpaqueElementsAttr` ( #1274 )
...
An earlier patch (bb47c166
2022-08-24 17:10:40 -05:00
f012279fa2
Add transposed case for at::convolution ( #917 )
...
Also adds a decomposition for aten::conv_transposed2d.input
2022-08-24 12:19:35 -04:00
3d0e18bbe7
Add decomposition for aten.roll ( #1170 )
...
* Add decomposition for aten.roll
* add e2e unittest
* refine type of torch.roll
* fix aten::cat output type
2022-08-24 08:36:05 +08:00
2374098d71
[MHLO] Init end to end unit tests ( #1223 )
2022-08-23 16:47:21 +08:00
65d811e267
[MLIR][TORCH] Fix dynamic cases for aten.index.Tensor
2022-08-19 12:13:20 +05:30
7bd173a1c4
[MHLO] Eliminate explicit dynamic output shape generating in converting AtenSliceTensorOp ( #1245 )
...
[MHLO] Eliminate explicit dynamic output shape generating in converting AtenSliceTensorOp
2022-08-19 10:14:57 +08:00
9be8997536
Revert "add native_dropout and related ops pattern ( #1211 )" ( #1230 )
...
This reverts commit c935795086
2022-08-17 13:48:10 +08:00
11a5b5ac52
[MHLO] Add AtenRSubScalarOp conversion pattern to MHLO ( #1233 )
...
* [MHLO] Add AtenRSubScalarOp conversion pattern
Co-authored-by: Bairen Yi <yibairen.byron@bytedance.com>
Co-authored-by: Jiawei Wu <xremold@gmail.com>
Co-authored-by: Tianyou Guo <tianyou.gty@alibaba-inc.com>
Co-authored-by: Xu Yan <yancey.yx@alibaba-inc.com>
Co-authored-by: Ziheng Jiang <ziheng.jiang@bytedance.com>
2022-08-17 09:07:36 +08:00
fde390c766
Re-enable custom op support
2022-08-16 22:49:08 +05:30
84d345c650
build: update llvm tag to 2dde4ba6 ( #1229 )
...
Summary of changes:
- Tensor dialect now sets `emitAccessorPrefix` to prefixed, thus
requring updates to methods that retrieve arguments
[https://reviews.llvm.org/D131361 ]
- Update MHLO to build with LLVM commit hash 2dde4ba6
- Replace `AbsOp` with `AbsFOp` [https://reviews.llvm.org/D131325 ]
- Replace deprecated `getValue()` with `value()`
[https://reviews.llvm.org/D131349 ]
- Remove `AnalysisState::defaultInitialize()`
[https://reviews.llvm.org/D131746 ]
- Update MHLO MLIR tests to use the updated assembly format
- Disabled two failing TOSA tests (Github Issue link:
https://github.com/llvm/torch-mlir/issues/1231 )
2022-08-15 23:54:45 -07:00
9d6ee48661
Fix unused-variables warnings about EmbeddingBag ops ( #1220 )
...
According to the documentation for
`torch.embedding_bag` (https://pytorch.org/docs/stable/generated/torch.nn.functional.embedding_bag.html ),
the default value for `scale_grad_by_freq` is False.
2022-08-15 09:43:55 -07:00
c935795086
add native_dropout and related ops pattern ( #1211 )
2022-08-15 09:28:47 +08:00
b1a506624c
Add decomposition of `aten.masked.tensor` op.
...
`aten.masked.tensor` op has been decomposed to `aten.masked.scalar` op.
2022-08-11 07:48:04 +05:30
d96ec64be1
remove torch dialect from legal list ( #1192 )
2022-08-11 09:22:41 +08:00
dd2da5a038
E2E support for AtenRemainderScalarOp ( #1200 )
2022-08-10 20:02:06 -04:00
738f4fe96a
Rename TorchToStd pass as TorchToArith ( #1163 )
...
All the converters in this pass appear to create ops from the
arith dialect. Hence the full rename.
Fix GH Issue #409 .
2022-08-10 20:12:51 +01:00
87562773f8
[MHLO] Add AtenCatOp conversion pattern to MHLO ( #1208 )
...
Co-authored-by: Bairen Yi <yibairen.byron@bytedance.com>
Co-authored-by: Jiawei Wu <xremold@gmail.com>
Co-authored-by: Tianyou Guo <tianyou.gty@alibaba-inc.com>
Co-authored-by: Xu Yan <yancey.yx@alibaba-inc.com>
Co-authored-by: Ziheng Jiang <ziheng.jiang@bytedance.com>
Co-authored-by: Vremold <xremold@gamil.com>
2022-08-09 22:12:34 -07:00
f83a905856
[MHLO]fix lowering failed on reduction op with i32 shape ( #1185 )
...
fixed lowering failed on torch::max.dim while shape type is i32
2022-08-09 17:02:50 +08:00
e55fc4deb5
Revert "E2E support for AtenRemainderScalarOp ( #1119 )" ( #1190 )
...
This reverts commit 34e207eeb5
2022-08-08 22:59:57 -07:00
bb47c166a0
llvm: update tag to 061e0189 ( #1180 )
...
Summary of changes:
- Switch to C++17 (similar to https://reviews.llvm.org/D131348 )
- Update MHLO to build with LLVM commit hash 061e0189
- Replace deprecated `hasValue()` and `getValue()` with `has_value()`
and `value()` respectively (https://reviews.llvm.org/D131349 )
- Use `TypedAttr` (https://reviews.llvm.org/D130092 )
- Use updated assembly format of `mhlo.compare` op (commit
d03ef01e70fbf9afd0fa1976fbb7ed31838929b3 in MHLO repo)
2022-08-08 20:17:35 -07:00
351f15424e
[MHLO] Add transposed convolution conversion pattern ( #1171 )
...
Co-authored-by: Bairen Yi <yibairen.byron@bytedance.com>
Co-authored-by: Jiawei Wu <xremold@gmail.com>
Co-authored-by: Tianyou Guo <tianyou.gty@alibaba-inc.com>
Co-authored-by: Xu Yan <yancey.yx@alibaba-inc.com>
Co-authored-by: Ziheng Jiang <ziheng.jiang@bytedance.com>
2022-08-09 09:50:07 +08:00
34e207eeb5
E2E support for AtenRemainderScalarOp ( #1119 )
...
* E2E support for AtenRemainderScalarOp
2022-08-08 20:02:52 -04:00
1ee865983b
[MHLO] fix tensor mode aten.div op pattern ( #1160 )
...
* [MHLO] fix tensor mode aten.div op pattern
See RFC #999
Co-authored-by: Bairen Yi <yibairen.byron@bytedance.com>
Co-authored-by: Jiawei Wu <xremold@gmail.com>
Co-authored-by: Tianyou Guo <tianyou.gty@alibaba-inc.com>
Co-authored-by: Xu Yan <yancey.yx@alibaba-inc.com>
Co-authored-by: Ziheng Jiang <ziheng.jiang@bytedance.com>
2022-08-06 23:38:06 +08:00
c94431f71c
[MHLO] Add convolution op pattern ( #1152 )
...
Co-authored-by: Bairen Yi <yibairen.byron@bytedance.com>
Co-authored-by: Jiawei Wu <xremold@gmail.com>
Co-authored-by: Tianyou Guo <tianyou.gty@alibaba-inc.com>
Co-authored-by: Xu Yan <yancey.yx@alibaba-inc.com>
Co-authored-by: Ziheng Jiang <ziheng.jiang@bytedance.com>
2022-08-04 00:41:35 -07:00
08fc2d89bb
Add non-unit groups support to aten.convolution ( #858 )
2022-08-04 02:18:38 -04:00
d030591df9
[MHLO] Init MHLO pooling-like op conversion ( #1141 )
...
* [MHLO] Init MHLO pooling-like op conversion and remove 'op' suffix in filenames
Co-authored-by: Bairen Yi <yibairen.byron@bytedance.com>
Co-authored-by: Jiawei Wu <xremold@gmail.com>
Co-authored-by: Tianyou Guo tianyou.gty@alibaba-inc.com
Co-authored-by: Xu Yan <yancey.yx@alibaba-inc.com>
Co-authored-by: Ziheng Jiang <ziheng.jiang@bytedance.com>
See RFC #999
2022-08-04 12:34:22 +08:00
f0a24f59f6
[MHLO] Init MHLO linear op patterns ( #1132 )
...
See RFC https://github.com/llvm/torch-mlir/issues/999
Co-authored-by: Bairen Yi yibairen.byron@bytedance.com
Co-authored-by: Jiawei Wu xremold@gmail.com
Co-authored-by: Tianyou Guo tianyou.gty@alibaba-inc.com
Co-authored-by: Xu Yan yancey.yx@alibaba-inc.com
Co-authored-by: Ziheng Jiang ziheng.jiang@bytedance.com
2022-08-03 19:10:54 -07:00
636f5acb10
[MHLO] Init MHLO reduce-like op conversion ( #1133 )
...
* [MHLO] init reduce-like op conversion from Torch to MHLO
Co-authored-by: Bairen Yi <yibairen.byron@bytedance.com>
Co-authored-by: Jiawei Wu <xremold@gmail.com>
Co-authored-by: Tianyou Guo <tianyou.gty@alibaba-inc.com>
Co-authored-by: Xu Yan <yancey.yx@alibaba-inc.com>
Co-authored-by: Ziheng Jiang <ziheng.jiang@bytedance.com>
2022-08-03 10:47:52 +08:00
0b23af27d3
[MHLO] support non-constant torch scalar in BasicOps ( #1134 )
...
See RFC https://github.com/llvm/torch-mlir/issues/999
Co-authored-by: Bairen Yi yibairen.byron@bytedance.com
Co-authored-by: Jiawei Wu xremold@gmail.com
Co-authored-by: Tianyou Guo tianyou.gty@alibaba-inc.com
Co-authored-by: Xu Yan yancey.yx@alibaba-inc.com
Co-authored-by: Ziheng Jiang ziheng.jiang@bytedance.com
2022-08-03 08:16:31 +08:00
38d8498b21
add e2e support for aten.atan2 ( #1117 )
...
- Includes math-to-libm pass in refbackend for math::atan2 support
2022-08-02 11:39:41 -04:00
704efdc259
[MHLO] add aten::gelu op pattern ( #1127 )
...
add aten::gelu op pattern, and moved some unit tests from basic.mlir to elementwise.mlir
2022-08-02 15:01:30 +08:00
76c976682c
[MHLO] Support for dynamic shape in basic op conversion by introducing CHLO dialect ( #1123 )
...
* [MHLO] Support for dynamic shape in basic op conversion by introducing CHLO dialect
Co-authored-by: Bairen Yi <yibairen.byron@bytedance.com>
Co-authored-by: Jiawei Wu <xremold@gmail.com>
Co-authored-by: Tianyou Guo <tianyou.gty@alibaba-inc.com>
Co-authored-by: Xu Yan <yancey.yx@alibaba-inc.com>
Co-authored-by: Ziheng Jiang <ziheng.jiang@bytedance.com>
* [MHLO] Support I32 as shape tensor dtype
* [NFC] Add a 'TODO' annotation
2022-08-02 12:53:24 +08:00
3772e0bd91
[NFC][MHLO] move util funcs to MhloLegalizeUtils.h/cpp ( #1128 )
...
See RFC: https://github.com/llvm/torch-mlir/issues/999
Co-authored-by: Bairen Yi yibairen.byron@bytedance.com
Co-authored-by: Jiawei Wu xremold@gmail.com
Co-authored-by: Tianyou Guo tianyou.gty@alibaba-inc.com
Co-authored-by: Xu Yan yancey.yx@alibaba-inc.com
Co-authored-by: Ziheng Jiang ziheng.jiang@bytedance.com
2022-08-02 09:21:37 +08:00
ed13ebfd8d
E2E support for AtenEmbeddingBagPaddingIdxOp SUM Mode ( #1066 )
2022-08-01 16:44:11 -04:00
c681c3497a
[MLIR][TORCH} Fix empty dim cases for the .dim ops
...
This commit fixes the shape calculation for:
1.) aten.mean.dim
2.) aten.var.dim
3.) aten.sum.dim_IntList op
Also, it fixes the lowering of `aten.mean.dim` and
`aten.sum.dim_IntList` for handling the cases of empty dim list.
Signed-Off By: Vivek Khandelwal <vivek@nod-labs.com
2022-07-29 11:08:57 +05:30
7247c6a3a7
[MLIR][TORCH] Add E2E support for aten.ge.int op
...
This commit adds lowering of `aten.ge.int` op.
Signed-Off By: Vivek Khandelwal <vivek@nod-labs.com>
2022-07-29 11:08:57 +05:30
11a8901078
[MLIR][TORCH] Add support for multiple indexing tensors for aten.index.Tensor ( #1097 )
...
- Includes a canonicalizer for `aten.add.t`needed for successfully lowering the shape function
- Only offers support for statically sized index tensors when there is more than one
- Dynamic shape support remains for single indexing tensors
2022-07-28 19:00:02 -04:00
3c9addf19c
Add e2e support for aten.expm1
2022-07-27 12:31:35 +05:30
052d2f84dc
[MHLO] Init MHLO basic op conversion ( #1092 )
...
* [MHLO] Init MHLO basic Op Conversion
Co-authored-by: Bairen Yi <yibairen.byron@bytedance.com>
Co-authored-by: Jiawei Wu <xremold@gmail.com>
Co-authored-by: Tianyou Guo <tianyou.gty@alibaba-inc.com>
Co-authored-by: Xu Yan <yancey.yx@alibaba-inc.com>
Co-authored-by: Ziheng Jiang <ziheng.jiang@bytedance.com>
* [NFC] Remove 'from @llvm-project' annotation
Co-authored-by: wujiawei.jw <wujiawei.jw@bytedance.com>
2022-07-27 13:07:51 +08:00
e8f327cc00
Add lowering to linalg for softplus and log1p
...
Follows existing conventions for unary operators.
2022-07-25 21:25:57 +05:30
44ead68772
[MHLO] Init MHLO gather op patterns ( #1104 )
...
See RFC https://github.com/llvm/torch-mlir/issues/999
Co-authored-by: Bairen Yi yibairen.byron@bytedance.com
Co-authored-by: Jiawei Wu xremold@gmail.com
Co-authored-by: Tianyou Guo tianyou.gty@alibaba-inc.com
Co-authored-by: Xu Yan yancey.yx@alibaba-inc.com
Co-authored-by: Ziheng Jiang ziheng.jiang@bytedance.com
2022-07-25 23:47:46 +08:00
f50d7013cd
[MHLO] Add [un]squeeze op patterns ( #1099 )
...
* [MHLO] Add [un]squeeze op patterns
* Conform to llvm coding standard
* minor update
2022-07-25 23:28:48 +08:00
b80ce79b9f
[MHLO] Init MHLO view like op patterns ( #1090 )
...
* [MHLO] Init MHLO view like op patterns
See RFC: https://github.com/llvm/torch-mlir/issues/999
Co-authored-by: Bairen Yi yibairen.byron@bytedance.com
Co-authored-by: Jiawei Wu xremold@gmail.com
Co-authored-by: Tianyou Guo tianyou.gty@alibaba-inc.com
Co-authored-by: Xu Yan yancey.yx@alibaba-inc.com
Co-authored-by: Ziheng Jiang ziheng.jiang@bytedance.com
* update filecheck test cases
* rebase, remove chlo and clang-format
2022-07-22 15:18:18 +08:00
a02dbb2d5e
[MHLO] Init MHLO slice like op patterns ( #1091 )
...
See RFC: https://github.com/llvm/torch-mlir/issues/999
Co-authored-by: Bairen Yi yibairen.byron@bytedance.com
Co-authored-by: Jiawei Wu xremold@gmail.com
Co-authored-by: Tianyou Guo tianyou.gty@alibaba-inc.com
Co-authored-by: Xu Yan yancey.yx@alibaba-inc.com
Co-authored-by: Ziheng Jiang ziheng.jiang@bytedance.com
2022-07-22 11:32:45 +08:00
c61c99e887
[MHLO] Init MHLO integration. ( #1083 )
...
Co-authored-by: Bairen Yi <yibairen.byron@bytedance.com>
Co-authored-by: Jiawei Wu <xremold@gmail.com>
Co-authored-by: Tianyou Guo <tianyou.gty@alibaba-inc.com>
Co-authored-by: Xu Yan <yancey.yx@alibaba-inc.com>
Co-authored-by: Ziheng Jiang <ziheng.jiang@bytedance.com>
2022-07-20 16:18:16 -07:00
647e75e029
Allow expanding and collapsing in aten::view ( #1082 )
...
- Supports cases where the view op expands and collapses dims
simulataneously. This does not handle the case where it is neither
expanding nor collapsing (e.g. [2, 3] -> [3, 2])
- Additionally fixes a previous bug with adding 1-sized dims on both
sides of a tensor with aten.view
2022-07-20 17:35:51 -04:00
c73a39e40a
Add support for index.Tensor on dimensions other than the first
...
This patch still only supports a single indexing tensor.
2022-07-19 11:36:52 +05:30
df0b1e77a4
[MLIR][TORCH] Add negative dim support for aten.cat and aten.slice op
...
This commit adds the support for negative dim cases for `aten.cat`,
`aten.slice.Tensor` and `aten.slice_scatter` op.
Signed-Off By: Vivek Khandelwal <vivek@nod-labs.com>
2022-07-18 14:01:33 +05:30
247dd64a66
Change to notifyMatchFailure ( #1073 )
...
emitError is intended for error cases and not match failures of
patterns. notifyMatchFailure is intended where pattern reports reason
for not matching.
Op verification should also not happen inside patterns but as part of
verify/verification, but left ones that were obviously verification to
emitError inside patterns to keep this change small.
2022-07-17 18:39:54 -07:00
Sean Silva
Ashay Rane
gpetters94
Tanyo Kwok
Vivek Khandelwal
武家伟
Yan Xu
nithinsubbiah
Ramiro Leal-Cavazos
Prashant Kumar
Vidush Singhal
Ramana Radhakrishnan
powderluv
Quinn Dawkins
Kevin Kiningham
Ziheng Jiang
Jacques Pienaar