14f1260ac4
Add more mhlo basic converters ( #1628 )
...
* Add more mhlo basic converters
* remove unused pinnedMemory constraints
* refine naming
2022-11-24 14:28:34 +08:00
4aad5ccf39
fix #1626 return type mismatch ( #1634 )
2022-11-23 15:02:41 +08:00
a9fb0c5459
fix mhlo e2e ci crashes ( #1620 )
...
* fix mhlo e2e ci crashes
* add passed tests
* calc dynamic positive dim
2022-11-21 21:50:35 +08:00
0d209998d1
llvm: update tag to e864ac6945 ( #1600 )
...
Summary of changes:
1. Replace `string` iterator types by `IteratorType` enum.
(e6598b053d9fd8d251a8
2022-11-16 14:40:36 -08:00
09ca07bca0
`m_TorchConstant{Int/Bool}List` -> `m_TorchListOfConstant{Int/Bool}s` ( #1601 )
...
This commit renames the patterns used to match on lists of constant
values to `m_TorchListOfConstant{valueType}s`. This is needed to avoid
ambiguity for when `valueType` has `Optional` in it. In particular, it
makes it clear whether the values in the list are optional or the list
itself is optional.
2022-11-16 20:33:12 +00:00
2793a2bd41
fix TorchToMhlo Conversion cmake dependency ( #1549 )
2022-11-09 18:34:53 -06:00
82a3860e25
build: update llvm tag to 4546397e ( #1502 )
...
This commit makes the following changes needed to update bump LLVM:
- Replace `linalg.init_tensor` with `tensor.empty` (see:
https://reviews.llvm.org/D135129 )
- Replace `NoSideEffect` with `Pure` (see
https://reviews.llvm.org/D135505 )
- Replace `body` region accessor for `ReduceOp` and `ReduceWindowOp`
with `getBody`
- Fix incorrect use of `tosa::ReduceSumOp` in `AtenNativeLayerNormOp`
conversion pattern. The result type of `tosa::ReduceSumOp` must have
the same rank as the input type. (see:
https://www.mlplatform.org/tosa/tosa_spec.html#_reduce_sum )
Co-authored-by: Ashay Rane <ashay@users.noreply.github.com>
Co-authored-by: Ashay Rane <ashay@users.noreply.github.com>
2022-10-18 04:22:53 +00:00
faa9a78e38
build: update llvm tag to 6f46ff37 ( #1448 )
...
Summary of changes:
- Updated references to the Arith dialect
(https://reviews.llvm.org/D134762 )
- Switched to prefixed accessors for MemRef dialect
(https://reviews.llvm.org/D134995 )
- Fixed warnings about signed/unsigned comparisons, ignored return
values, and unused variables
2022-10-05 08:28:06 -05:00
a60acf272d
build: update llvm tag to bebc9695 ( #1415 )
...
Summary of changes:
- Renamed OptionalArrayRefParameter since the name conflicts with an
upstream symbol that has a different meaning
(https://reviews.llvm.org/D133819 )
- Removed extraneous dependency between TorchMLIRTorchToMhlo and
ChloOps, since the existing dependency on MhloDialect is sufficient
- Fixed code to prevent warnings related to comparisons between signed
and unsigned values
2022-09-26 11:44:54 -05:00
ab7aa01b1e
[MHLO] Add torch-to-mhlo e2e support for aten.gather op ( #1410 )
...
* Add torch-to-mhlo e2e support for aten.gather op
* Add more e2e tests for torch.aten.gather op
2022-09-25 22:07:46 +08:00
16dd7e2e5f
Fix dynamic shapes type verifications ( #1409 )
...
* Fix dynamic shapes type verifications
2022-09-23 20:50:29 +08:00
72e422b589
Add relu6 and binary broadcasts ( #1408 )
...
* Add relu6 and binary broadcasts
2022-09-23 20:39:15 +08:00
061a97c3f2
Replace empty_like && empty_memory_format with full/full_like ( #1398 )
...
* Replace empty_like && empty_memory_format with full/full_like
* fix broadcast rank0 tensor
2022-09-23 10:24:36 +08:00
0e2e94d542
Add torch-to-mhlo e2e support for AtenArangeStartStepOp ( #1385 )
...
Co-authored-by: Vremold <xremold@gamil.com>
2022-09-20 22:31:24 +08:00
b316918947
Add AtenClampOp conversion pattern to MHLO ( #1356 )
...
Add AtenClampOp conversion pattern to MHLO
2022-09-16 15:09:21 +08:00
6a1893a517
[MLIR][MHLO] Add AtenFrobeniusNormDimOp and add its conversion pattern to MHLO and linalg ( #1306 )
...
* Add aten.frobenius_norm.dim op and init its conversion pattern to linalg and MHLO,
* run symbolic-shape-optimization before hlo-legalize-to-linalg to fit more mhlo e2e tests.
2022-09-08 10:15:36 +08:00
93f7c0ceb5
build: update llvm tag to d2613d5b ( #1343 )
...
Summary of changes:
- Update the dataflow analysis in RefineTypes.cpp
- Add tosa-to-arith pass after tosa-to-linalg pass, since
tosa-to-linalg (and canonicalizations) can produce tosa.const() ops
- Fixed warning about not making `matchAndRewrite` as override
2022-09-07 14:35:14 -05:00
29cafdbb61
[MHLO] refactor pass configurations ( #1315 )
...
Related to https://github.com/llvm/torch-mlir/issues/1227
1. Reduce MHLO #ifdefs
2. Dismiss compilation warnings
2022-09-01 10:36:02 +08:00
e52e886845
build: update llvm tag to 00d648bd ( #1307 )
...
- Update MHLO commit to build with LLVM commit hash 00d648bd
- Update TorchToMhlo code to work with Stablehlo
- Re-enabled two failing TOSA tests, thus resolving Github Issue #1231
2022-08-30 14:44:00 -05: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
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
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
c935795086
add native_dropout and related ops pattern ( #1211 )
2022-08-15 09:28:47 +08:00
d96ec64be1
remove torch dialect from legal list ( #1192 )
2022-08-11 09:22:41 +08: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
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
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
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
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
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
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
Tanyo Kwok
Gaurav Shukla
Ramiro Leal-Cavazos
Yuanqiang Liu
Ashay Rane
武家伟
Yan Xu
Ziheng Jiang