e45ad313d4
build: update llvm tag to 147fe9de
...
Summary of changes:
- Update call to `hasNoEffect` utility
- `KDynamicSize` value changed to
`std::numeric_limits<int64_t>::min()` from `-1`
- Update tags
llvm: 147fe9de29dc13c14835127b35280c4d95c8e8ba
mhlo: 1944b5fa6062ec4c065d726c9c5d64f1487ee8c5
Signed-Off By: Vivek Khandelwal<vivek@nod-labs.com>
2022-11-24 12:44:43 +05:30
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
bfcfd60d55
[MLIR][TORCH] Refix differentiable view ( #1639 )
...
* `BatchMlpLayerModule_basic` passes
* Fix https://github.com/llvm/torch-mlir/issues/1618 by stripping `requires_grad` from results of view ops.
2022-11-23 15:35:39 -06:00
4aad5ccf39
fix #1626 return type mismatch ( #1634 )
2022-11-23 15:02:41 +08:00
68f568b704
[MLIR][TORCH] Add E2E support for prims.convert_element_type op
...
Signed-Off By: Vivek Khandelwal<vivek@nod-labs.com>
2022-11-22 09:36:36 +05:30
22307a1427
Clean up some parts of the test suite
...
The purpose of the test suite is to accelerate the development of the
compiler. However, we had various tests there that were not expected to
work, had no in-progress work being tested by the test, and nobody was
actively working on them. Having such tests in our test suite just adds
clutter and slows down development on the compiler.
2022-11-21 06:14:31 -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
1d949f3ac2
[MLIR][TORCH] Fix aten.upsample_nearest2d op
...
-- aten.upsample_nearest2d.vec op is not present
owing to https://github.com/pytorch/pytorch/pull/85638
-- So this commit adds a lowering on aten.upsample_nearest2d.
Signed-off-by: Abhishek Varma <abhishek@nod-labs.com>
2022-11-18 13:41:47 +05:30
5f7177da35
[MLIR][TORCH] Add decomposition for aten.var_mean.correction op
...
Signed-Off By: Vivek Khandelwal<vivek@nod-labs.com>
2022-11-17 13:00:09 +05:30
92f385bd9f
[MLIR][TORCH] Add E2E support aten.convolution_backward op
...
This commit adds the decomposition for the `aten.convolution_backward`
and `aten.convolution_backward_overrideable` op.
2022-11-15 07:38:26 +05:30
dfe7513a45
[MLIR][TORCH] Fix aten.unsqueeze op ( #1578 )
...
The range of the unsqueeze dim is: [-input.dim() - 1, input.dim() + 1), the bug forgets to add 1.
2022-11-14 09:09:15 -08:00
6909eaf7fc
Update TorchMlirBackendImpl Methods ( #1580 )
...
* Fix LTC build
* Remove passing test from xfail set
2022-11-14 00:37:49 -05:00
a7ac0def45
Move single-tensor-tuple-return test to mlir unit test.
...
Also, add multiple return test.
2022-11-10 09:23:53 -05:00
fedf8c0640
[MLIR][TORCH] Add E2E support for aten.upsample_nearest2d_backward.vec op
...
Signed-Off By: Vivek Khandelwal<vivek@nod-labs.com>
2022-11-04 22:10:07 +05:30
c86177730d
[MLIR][TORCH] Add E2E support for aten.fill.Tensor op
...
This commit adds the decomposition for `aten.fill.Tensor` op.
Signed-Off By: Vivek Khandelwal <vivek@nod-labs.com>
2022-10-30 18:40:47 +05:30
ea602127b6
[MLIR][TORCH] Add E2E support for aten.addcmul_ and aten.addcdiv_ op
...
Signed-Off By: Vivek Khandelwal <vivek@nod-labs.com>
2022-10-28 16:07:50 +05:30
3e199aaf11
Add better error message for single-tensor tuple returns.
2022-10-25 12:48:55 -04:00
ca87033d2f
[MLIR][TORCH] Add E2E support for aten.mse_loss op
...
This commit adds decomposition for the `aten.mse_loss` op.
Signed-Off By: Vivek Khandelwal <vivek@nod-labs.com>
2022-10-25 21:06:58 +05:30
ad6f5848cb
[MLIR][TORCH] Add TorchToTosa lowering for aten.where.self op ( #1454 )
2022-10-18 09:39:39 -07: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
bdb5083d33
New ops support & enhancements ( #1494 )
...
* New ops support & enhancements
* Enabled xfail ltc tests
2022-10-14 10:28:21 -04:00
3a2cd23380
[LINALG] Add lowering for aten::round op.
...
-- Added the lowering for aten::round op.
-- Added the folding for integer cases.
2022-10-13 02:41:26 +05:30
aefbf65e27
Disable LTC and update PyTorch ( #1472 )
...
* build: disable LTC again so that we can bump PyTorch version
When built using PyTorch's master branch, the LTC code has been failing
to build for a few days. As a result, the PyTorch version referenced by
Torch-MLIR is stalled to the one from October 4th.
In an effort to advance to PyTorch version, this patch disables LTC, and
a subsequent patch will advance the PyTorch version.
* update PyTorch version to 1.14.0.dev20221010
Also disables the `UpSampleNearest2dDynamicFactor_basic` e2e test, since
the (PyTorch) oracle differs from the computed value for both the
refbackend and the eager_mode backends.
2022-10-10 23:05:40 -05:00
da90a25f90
[MLIR][TORCH] Add E2E support for `aten.[div.int|bitwise_or.Tensor]` ops
...
This commit adds lowering of `aten.div.int` and `aten.bitwise_or.Tensor`
ops. Both these ops are required in order to support bloom_560m model.
Signed-Off-by: Gaurav Shukla <gaurav@nod-labs.com>
2022-10-10 22:28:51 +05:30
d3cc3f1aff
[tosa] Add lowering for aten.to.dtype and aten._to_copy op
...
This commit adds the TorchToTosa lowering for `aten.to.dtype` and
`aten._to_copy` op.
Signed-Off By: Vivek Khandelwal<vivek@nod-labs.com>
2022-10-06 12:00:25 +05:30
56f9a9b5de
[tosa] Add TorchToTosa lowering for torch.prim.NumToTensor.Scalar op
...
Signed-Off By: Vivek Khandelwal<vivek@nod-labs.com>
2022-10-06 12:00:25 +05:30
2ba71af651
Add support for mv decomposition.
2022-10-04 11:34:45 -04:00
6777a9484d
[LINALG] Add lowering for the aten.upsample_nearest2d op.
2022-10-04 17:20:29 +05:30
9dd5ae8239
[tosa] Add TorchToTosa lowering for aten.arange.start_step op ( #1442 )
2022-09-30 07:33:41 -07:00
940959589b
[MLIR][TORCH] Add Byte and Char Dtype support
2022-09-30 13:19:31 +05:30
6db513c51d
[tosa] Add support for some cases of aten.broadcast_to op ( #1429 )
...
This commit adds support for TorchToTosa lowering of
`aten.broadcast_to` op for cases:
1.) When the rank of input and output tensor is equal.
2.) When the rank of input tensor is zero.
Signed-Off By: Vivek Khandelwal<vivek@nod-labs.com>
2022-09-29 09:40:56 -07:00
8ef0c874c2
Implement Expand/Collapse Functionality for Aten.View ( #1353 )
2022-09-27 11:08:14 -07:00
3e27aa2be3
Fix as_strided/slice symint ( #1401 )
...
* Fix as_strided symint
* Re-enable LTC tests
* Re-enable LTC
* Add hardtanh shape inference function
* Fix slice symint
2022-09-26 12:16:49 -04: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
bc11e1aba6
[tosa] Add "-tosa-to-tensor" pass in the lowering pipeline
...
Signed-Off By: Vivek Khandelwal<vivek@nod-labs.com>
2022-09-24 10:03:07 +05:30
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
4ef6e69ed4
[MLIR][TORCH] Add TorchToTosa lowering for aten.clone op ( #1388 )
...
Signed-Off By: Vivek Khandelwal <vivek@nod-labs.com>
Co-authored-by: Suraj Sudhir <16977902+sjarus@users.noreply.github.com>
2022-09-20 15:07:46 -07:00
5090ac9359
[MLIR][TORCH] Add a test for sum.dim_IntList op working for tosa ( #1387 )
...
Signed-Off By: Vivek Khandelwal<vivek@nod-labs.com>
Co-authored-by: Suraj Sudhir <16977902+sjarus@users.noreply.github.com>
2022-09-20 11:38:09 -07:00
1ffd42bbde
[MLIR][TORCH] Add TorchToTosa lowering for aten.broadcast_to op ( #1386 )
...
Signed-Off By: Vivek Khandelwal<vivek@nod-labs.com>
2022-09-20 10:04:51 -07: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
51e3c3f1ed
[MLIR][TORCH] Add failing test to xfail_sets.py
...
Signed-Off By: Vivek Khandelwal <vivek@nod-labs.com>
2022-09-16 17:12:21 +05:30
b316918947
Add AtenClampOp conversion pattern to MHLO ( #1356 )
...
Add AtenClampOp conversion pattern to MHLO
2022-09-16 15:09:21 +08:00
48418b9c22
Fold away type_as ( #1358 )
2022-09-12 18:59:12 -04:00
71b1f0dd7a
[MLIR][TORCH] Add E2E support for aten.index.Tensor_hacked_twin op
...
This commit adds lowering of `index.Tensor_hacked_twin` op.
Signed-Off By: Vivek Khandelwal <vivek@nod-labs.com>
2022-09-12 21:47:18 +05:30
e35741fb1d
[MLIR][TORCH] Add E2E support for aten.bitwise_not op
...
This commit adds lowering of `aten.bitwise_not` op.
Signed-Off By: Vivek Khandelwal <vivek@nod-labs.com>
2022-09-08 17:52:12 +05:30
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
cc86cc0f02
Revert "Implement Non-Expand/Collapse Functionality for Aten.View ( #1309 )" ( #1347 )
...
Reverting commit a6a48ba233#1309
2022-09-07 01:38:11 -04:00
a6a48ba233
Implement Non-Expand/Collapse Functionality for Aten.View ( #1309 )
...
Focuses on statically sized cases such as [2, 3] -> [3, 2].
2022-09-06 14:46:04 -04: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
Vivek Khandelwal
Tanyo Kwok
Maksim Levental
Sean Silva
Abhishek Varma
George Petterson
Chi_Liu
Gleb Kazantaev
Daniel Ellis
Chi_Liu
Ramiro Leal-Cavazos
Prashant Kumar
Ashay Rane
Gaurav Shukla
AmosLewis
JakopinA
Jae Hoon (Antonio) Kim
武家伟
Quinn Dawkins