d61986cfcf
Add Decompostion for `Aten_SafeSoftmaxOp` ( #3708 )
...
Co-authored-by: Vivek Khandelwal <vivekkhandelwal1424@gmail.com>
2024-09-12 16:58:10 -05:00
edf725ef42
[Torch] add AtenAsStridedOp in torch dialect ( #3706 )
2024-09-12 19:07:11 +08:00
1c4b9d6a0e
Implement lowering of torch.aten.hstack ( #3563 )
2024-09-11 16:41:47 +05:30
70d5730c87
[LINALG] Implement lowering of torch.aten.rot90 ( #3551 )
2024-09-06 10:36:17 +05:30
567ed44fd0
[MLIR][TORCH] Add E2E support for aten.polar op ( #3671 )
...
Signed-Off By: Vivek Khandelwal <vivekkhandelwal1424@gmail.com>
2024-09-03 10:51:03 +05:30
5bc59ce1fa
[TorchToLinalg] Support lowering MaxPool3dWithIndices ( #3652 )
...
Support torch.MaxPool3dWithIndices lowering to linalg backend.
2024-08-27 14:14:25 -05:00
0a86deb59a
build: manually update PyTorch version ( #3627 )
...
Set PyTorch and TorchVision version to nightly release 2024-08-18.
This commit also updates the `scaled_dot_product_attention` op.
A new attribute `enable_gqa` has been added. As of now, only the
default value for the same is supported.
Signed-Off By: Vivek Khandelwal <vivekkhandelwal1424@gmail.com>
2024-08-19 12:03:56 +05:30
23ec5399e5
Implement lowering of aten.atleast_2d ( #3546 )
...
This operator is needed to implement aten.vstack, which will be
submitted in a subsequent PR
2024-08-14 18:52:31 +05:30
10fe5d08d1
Implement lowering for torch.aten.rad2deg ( #3586 )
2024-08-14 16:37:28 +05:30
c5b3cf299a
[Torch] emit upsample_nearest1d/2d/vec, and add shape/dtype functions ( #3629 )
2024-08-13 19:14:24 +08:00
8bd1b9751f
`max_unpool3d` linalg lowering ( #3536 )
...
An attempt of `aten.max_unpool3d` to linalg lowering.
There are known issues with this implementation (see comment in code).
2024-07-30 20:59:17 +03:00
b6e4725259
[ONNX] Add OnnxToTorch lowering for NonMaxSuppression op ( #3501 )
...
Signed-Off By: Vivek Khandelwal <vivekkhandelwal1424@gmail.com>
2024-07-26 21:01:27 +05:30
d1e172f418
Register fake_quantize_cachemask ops and add their decompose patterns ( #3556 )
...
Test:
`cmake --build build --target check-torch-mlir-all`
2024-07-23 11:33:12 -07:00
22c9008bb9
build: Update Roll PyTorch version ( #3548 )
...
This commit also updates the PyTorch and Torchvision nightly links since
they are now moved to a different location.
PyTorch Nightly: https://download.pytorch.org/whl/nightly/cpu/torch/
Torchvision Nightly:
https://download.pytorch.org/whl/nightly/cpu/torchvision/
Disables dtype checks for some ops, tracked by https://github.com/llvm/torch-mlir/issues/3552
Signed-Off By: Vivek Khandelwal <vivekkhandelwal1424@gmail.com>
2024-07-19 21:38:57 +05:30
2cdf3deae3
implement lowering of torch.aten._linalg_slogdet ( #3524 )
2024-07-19 11:24:43 +05:30
c7d972ed58
Implement lowering of torch.aten.tril_indices ( #3517 )
2024-07-18 18:38:12 +05:30
fde286f491
Implement lowering for torch.aten.hann_window.periodic ( #3502 )
2024-07-17 18:21:23 +05:30
b59efc75f3
Implement lowering of torch.aten.atleast_1d ( #3498 )
...
This operator is necessary in order to implement torch.aten.vstack.
Which will be added in a future PR.
2024-07-17 18:20:30 +05:30
574143448b
[E2E][ONNX] torch.multinomial ( #3404 )
...
This PR adds a conversion in the TorchOnnxToTorch pass for the ONNX
Multinomial operation. It also adds a TorchToLinalg lowering for the
`aten.Multinomial` op and does a light refactor of some repeated code
that generates random floating point numbers in
`TorchToLinalg/Random.cpp`.
2024-07-16 23:09:39 +05:30
0791a8860c
[Torch] Implements TorchToLinalg lowering of torch.ops.aten._weight_norm_interface ( #3538 )
...
Resolves https://github.com/nod-ai/SHARK-Turbine/issues/757 .
Adds TorchToLinalg lowering for `Aten_WeightNormInterfaceOp`.
---------
Co-authored-by: Ubuntu <rbhowmik@RohanBhowmikVM.judsoscro3wupi0qm4bjlj5m3b.bx.internal.cloudapp.net>
2024-07-16 23:09:12 +05:30
5e4f00acb1
[Torch] add support for aten.scatter_add ( #3534 )
2024-07-12 09:15:42 +08:00
d466d5b809
Register fake_quantize related ops ( #3522 )
...
Register `aten.fake_quantize_per_channel_affine` and
`aten.fake_quantize_per_tensor_affine.tensor_qparams` ops
---------
Co-authored-by: Ze Zhang <ze.zhang@getcruise.com>
2024-07-05 11:02:03 -07:00
e2fbded49c
[Torch Dialect] improve argmax/argmin's decomposition to support keep… ( #3514 )
...
…dim=True when dim=None
2024-07-02 09:08:57 +08:00
0e71a192d8
[Torch] support decomposition of aten.aminmax ( #3513 )
...
* unify decompisition of `aten.amax` and `aten.amin`
* support `aten.amax` with `dim=()`
2024-06-29 21:44:05 +08:00
f75cbb4df9
[torch dialect] emit aten.fmax/fmin and add decomposition patterns ( #3510 )
2024-06-29 00:07:55 +08:00
5a627c46b7
onnx.DFT basic support ( #3463 )
...
- adds support for DFT v20 on the FFT and IFFT path
- adds required skeleton code for IFFT ops to be recognised in TMlir
2024-06-28 20:08:43 +05:30
d2bc70f188
[TorchToLinalg][ONNX] Add Basic Determinant Support ( #3481 )
...
This adds support for a few ops:
- torch.linalg_det
- torch._linalg_det (if the LU and pivot returns are unused)
- onnx.Det
An scf loop is used, since the row reduction algorithm applied here has
some loop-carried dependencies.
The current support being added here is very basic, and only works if no
permutations are required during row reduction, and assumes the matrices
are non-singular.
2024-06-25 13:34:19 -05:00
368fabf0c1
[ONNX] Basic Support for DeformConv ( #3469 )
...
This adds a torchvision op to torch-mlir and a path from onnx.DeformConv
to torchvision.deform_conv2d.
I'm not implementing the torch->linalg lowering for the torchvision op
yet, but posting this PR to get feedback on some of the choices being
made here and to flesh out the onnx frontend a bit.
2024-06-25 12:16:51 -05:00
e346c911f7
[ONNX] Add basic support for RoiAlign ( #3493 )
...
This adds an onnx->torch conversion for onnx.RoiAlign into
torchvision.roi_align or torchvision.roi_pool, and adds those two
torchvision ops to torch-mlir.
2024-06-25 11:02:45 -05:00
02340408b7
[torch] Add OnnxToTorch lowering for Onnx.STFT op ( #3492 )
...
Adds OnnxToTorch lowering for `Onnx.STFT` op.
2024-06-25 19:00:45 +05:30
98c6971a01
Implement lowering of torch.aten.triu_indices ( #3451 )
...
Closes
[nod-ai/SHARK-Turbine/issues/709](https://github.com/nod-ai/SHARK-Turbine/issues/709 )
---------
Co-authored-by: Branko Trifkovic <branko.trifkovic@syrmia.com>
2024-06-21 16:16:38 -07:00
acd57a3520
Support fake_quantize_per_tensor_affine_cachemask ( #3477 )
...
Add a new op with shape/dtypes and decompose into
`fake_quantize_per_tensor_affine` when the second result is unused.
The xfail_set change is on ONNX because torch cannot export this op to
ONNX.
2024-06-21 07:15:31 +00:00
c7d52f63b4
[stablehlo] add aten::_int_mm lowering ( #3474 )
...
as title
2024-06-20 16:10:31 +08:00
676fa8cc09
Implement lowering of torch.aten.renorm ( #3388 )
...
Closes
[nod-ai/SHARK-Turbine/issues/689](https://github.com/nod-ai/SHARK-Turbine/issues/689 )
---------
Co-authored-by: Branko Trifkovic <branko.trifkovic@syrmia.com>
2024-06-17 10:40:57 -07:00
4555629246
Implement lowering of torch.aten.kthvalue ( #3360 )
...
Closes
[nod-ai/SHARK-Turbine#620 ](https://github.com/nod-ai/SHARK-Turbine/issues/620 )
2024-06-15 11:18:39 +05:30
39d882f7c9
[torch] Add OnnxToTorch lowering for the Col2Im op ( #3424 )
...
Adds OnnxToTorch lowering for the `onnx.Col2Im` op.
2024-06-13 08:42:06 +00:00
72837fbb3d
build: manually update PyTorch version ( #3340 )
...
Set PyTorch and TorchVision version to nightly release 2024-05-14.
Signed-Off By: Vivek Khandelwal <vivekkhandelwal1424@gmail.com>
2024-06-06 22:23:40 +05:30
661be2d5b0
[MLIR][Torch] Add TorchToLinalg lowering for AtenAvgPool3dOp ( #3030 )
...
This commit also fixes the average pool op' test failing for
OnnxToLinalg lowering.
Signed-Off By: Vivek Khandelwal <vivekkhandelwal1424@gmail.com>
2024-06-04 22:12:34 +05:30
285b087a5d
[Torch] Emit rrelu and decompose it ( #3250 )
...
as title
2024-06-03 19:25:52 +08:00
23b53050de
[Torch]Support conv_transpose1d and conv_transpose3d ( #3286 )
...
1. Support conv_transpose1d and conv_transpose3d
2. Fix bugs of convertTransposedConv func in
lib/Conversion/TorchToStablehlo/Linear.cpp
2024-06-03 15:11:12 +08:00
5bb1a65ec9
[Stablehlo] refactor reduction lowering and support aten.amin ( #3383 )
...
* implement detailed lowering template pattern
`ConvertAtenReduceAllDimsOp` and `ConvertAtenReduceKeepDimOp`
* support `aten.amin`'s lowering.
2024-05-23 20:40:20 +08:00
8814d0ae64
[Torch] emit aten.dot and canonicalize it to aten.matmul ( #3361 )
...
* canonicalize `aten.dot` to `aten.matmul`
2024-05-18 22:45:14 +08:00
1d4859699b
MaxPool1d lowering to linalg ( #3295 )
...
Co-authored-by: root <root@i32b01216.sqa.eu95>
2024-05-10 22:05:26 +05:30
ec6d7aa5d2
OnnxToTorch lowering resize op ( #3013 )
...
https://github.com/nod-ai/SHARK-Turbine/issues/358
adds a lowering from onnx to linalg for bilinear and nearest resize with
support for using scales or sizes to get resize shape. uses coordinate
transform half pixel for bilinear mode and asymmetrical for nearest
mode. See
https://github.com/onnx/onnx/blob/main/docs/Operators.md#Resize . Added
two passes -- one for bilinear and the other for nearest.
2024-05-08 21:35:03 +00:00
abef114c0c
[torch] emit aten.Softshrink and aten.Hardshrink ( #3248 )
...
as title
2024-05-08 15:20:45 +08:00
f32ada993d
[Stablehlo] Improve the lowering of pool op in stablehlo ( #3259 )
...
1. Handle case stride == None
2. add avgpool3d maxpool1d maxpool3d lowering
2024-05-01 00:06:13 +08:00
5684dc0441
[Torch] emit aten.celu and decompose it ( #3247 )
...
CELU(x)=max(0,x)+min(0,α∗(exp(x/α)−1))
2024-04-28 17:23:40 +08:00
46c0f3cad0
[Torch] emit aten.log_sigmoid and decompose it to log(sigmoid) ( #3246 )
2024-04-28 11:47:43 +08:00
fab2696489
[Torch] support aten.trunc ( #3219 )
...
decompose `trunc(x)` to `sign(x) * floor(abs(x))`
2024-04-24 14:32:33 +08:00
4da3d714cc
[Torch] Support AtenProdOp on linalg and stablehlo ( #3215 )
2024-04-24 11:14:04 +08:00
zjgarvey
yyp0
Branko Trifkovic
Vivek Khandelwal
lingzhiz1998
pkapris-syrmia
Yuanqiang Liu
Ivan Butygin
Ze Zhang
bosko-syrmia
Arham Khan
rohan-tan-bhowmik
Jiawei Wu
Phaneesh Barwaria
Vinayak Dev
Matthias Gehre
Xinyu Yang
ptrifunovic98
NeverRaR
aldesilv