a2bfe47faa
[onnx] Add IDF and TFIDF modes to TFIDF Vectorizer ( #3726 )
...
Address https://github.com/nod-ai/SHARK-Turbine/issues/833
2024-10-02 08:17:58 -05:00
a33d1232c5
[onnx] Fix onnx.Shape lowering with scalar input ( #3716 )
...
Address https://github.com/nod-ai/SHARK-Turbine/issues/826
2024-09-27 13:30:02 -07:00
99848265c3
[onnx] Relax constraints on input tensors in `onnx.STFT` conversion to torch dialect ( #3676 )
...
- When the signal tensor is real, onnx allows its shape to be
`[batch][length]` as well as `[batch][length][1]`.
- Onnx also allows to specify `frame_length` together with `window` (not
empty), given that it matches the window size.
- Adding checks on signal and result shapes.
2024-09-23 12:09:29 +05:30
3a599bec80
[onnx] Fix onnx.ThresholdedRelu crash ( #3638 )
...
Result type was not fetched causing a crash on construction
2024-08-16 09:23:38 -07:00
a4ba02eef5
[ONNX] add support for tfidfvectorizer ( #3553 )
...
1-d/2-d input and output
implemented based on the description and example test cases in
https://github.com/onnx/onnx/blob/main/docs/Operators.md#TfIdfVectorizer
and some notes from
https://github.com/onnx/onnx/blob/main/onnx/reference/ops/op_tfidf_vectorizer.py#L128
---------
Co-authored-by: zjgarvey <zjgarvey@gmail.com>
2024-08-12 18:10:11 -05:00
44266ab0c4
[onnx] Support `fp8` for `onnx.QuantizeLinear` ( #3619 )
...
We need to directly decompose quantize linear for `fp8` types as the
equivalent torch operations do not support the operation.
2024-08-09 12:32:46 -07:00
880e64bbbb
[onnx] `onnx.Split` may not have `num_outputs` which can be inferred ( #3608 )
...
The attribute does not exist in all variants of the operation. It can be
inferred from the number of results so we should just do that.
2024-08-08 16:17:38 -07:00
341f415b1e
[onnx] Fix lowering `onnx.Shrink` to Torch ( #3603 )
...
This fixes the result type of the `torch.aten.lt.Scalar` and
`torch.aten.ge.Scalar` ops created during the lowering of `onnx.Shrink`
to Torch.
2024-08-07 21:25:14 +02:00
18139994e8
[onnx] Fix edge condition for `onnx.ReduceMax` ( #3598 )
...
For length-0 on `onnx.ReduceMax` the length 0 case was incorrect due to
a copy paste error.
2024-08-07 10:32:28 -07:00
b1a232222f
[onnx] Fix `onnx.Shape` to include `start` and `end` processing ( #3580 )
...
`onnx.Shape` can select only a subset of indices using attributes. Add
support for these attributes.
---------
Co-authored-by: zjgarvey <47986913+zjgarvey@users.noreply.github.com>
2024-08-05 13:56:07 -07:00
839fe90f86
[MLIR][ONNX] Add support for onnx.scan op ( #3516 )
...
This commit lowers onnx.scan op to torch.prim.Loop op and adds the
lowering in the onnx pipeline.
Signed-off-by: Gaurav Shukla <gaurav.shukla@amd.com>
2024-08-05 15:37:26 +05:30
3d33c5a206
[onnx] Fix `onnx.ScatterElements` for negative indices ( #3582 )
...
We need to adjust for negative scatter indice values. Added
materializing out the inbounds adjustment.
2024-08-02 09:01:10 -07:00
306ed62edd
[onnx][torch] Fix `onnx.SoftmaxCrossEntropyLoss` for ignore index ( #3585 )
...
There were two issues related to `ignore_index` being set
(1) the onnx-to-linalg pass as not reading the value correctly (2) the
mean pass was not considering the `ignore_index` value
For (2) when taking the mean we need to know how many of the values were
considered in the sum and therefore we cannot divide by the total number
of elements. Adding a summation across the total number should correct
this issue.
2024-08-02 09:00:56 -07:00
30c4d2f2b8
[torch] Add OnnxToTorch lowering for Onnx.Unique op ( #3523 )
...
Adds OnnxToTorch Lowering for the `Onnx.Unique` op.
2024-07-29 17:32:44 +05:30
a211ccbcff
Implementation of SplitToSequence ops lowering ( #3509 )
...
Added support for splitToSequence ops lowering
Added test case with filecheck
2024-07-29 15:44:22 +05:30
f0ce1e94ce
[ONNX] Add OnnxToTorch support for SequenceMap ( #3535 )
2024-07-17 14:25:09 -07:00
5342aa70cf
Support onnx.GRU and onnx.RNN ( #3447 )
2024-07-10 14:04:17 -04:00
0b46d1110a
[MLIR][ONNX] Add support for onnx.ScatterND ( #3479 )
...
This commit adds support for onnx.ScatterND op in the onnx pipeline.
Signed-off-by: Gaurav Shukla <gaurav.shukla@amd.com>
2024-07-08 13:27:14 +05:30
6d0ca499e6
[ONNX] Add OnnxToTorch support for ReverseSequence ( #3495 )
2024-06-27 14:33:41 -07: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
3c3fbe4680
[ONNX] Add OnnxToTorch lowering for Onnx.Upsample Op ( #3371 )
...
Signed-Off By: Vivek Khandelwal <vivekkhandelwal1424@gmail.com>
2024-06-25 12:58:31 +05:30
41d04a8995
[onnx] Resize supports default-valued attributes ( #3450 )
...
Handles onnx exporters emitting default-valued attributes.
Signed-off-by: Suraj Sudhir <suraj.sudhir@arm.com>
2024-06-12 09:23:42 -07:00
de28c8540b
[ONNX] add int16 quantization support ( #3446 )
...
There is currently no int16 quantization support in torch. This patch
adds a new mlir type to correspond to the missing "torch.qint16" type,
and enables lowering of quantization-related onnx ops using int16 types.
In follow-up patches, custom quantization logic for ops like
aten.matmul/aten.mm/aten.convolution may need to be revisited to allow
support for qint16. The passes in FuseQuantizedOps.cpp may also need
slight modifications.
2024-06-12 10:37:22 +05:30
e07a0bfc54
onnx.resize: Add support for coordTfMode "half_pixel" ( #3441 )
...
half_pixel is also the default mode used by ONNX, see
https://onnx.ai/onnx/operators/onnx__Resize.html
2024-06-10 20:59:29 +02:00
d35b6b412a
[ONNX] Add OnnxToTorch Lowering for Sequence Ops ( #3425 )
...
This commit adds the lowering for SequenceAt, SequenceEmpty,
SequenceInsert, SequenceErase op
Signed-Off By: Vivek Khandelwal<vivekkhandelwal1424@gmail.com>
2024-06-08 09:58:11 +05:30
f794582b18
add resize nearest mode round_prefer_floor, round_prefer_ceil, ceil ( #3421 )
2024-06-07 14:04:11 -05:00
1a9c0a35a9
[Onnx] Add Onnx->Torch lowering for Onnx.Shrink Op ( #3385 )
...
Signed-Off By: Vivek Khandelwal <vivekkhandelwal1424@gmail.com>
2024-06-07 22:47:27 +05:30
6382dbbcc0
[ONNX] Add OnnxToTorch lowering for SpaceToDepth op ( #3393 )
...
Signed-Off By: Vivek Khandelwal <vivekkhandelwal1424@gmail.com>
2024-06-03 20:29:39 +05:30
617b00b983
[NFC] Fix member cast change to global for landing collision ( #3407 )
...
A PR landed when moving away from a deprecated cast function. Updated
the corresponding lines to pass.
2024-05-31 17:31:24 +00:00
fc100a117d
[MLIR][ONNX] Add OnnxToTorch support for Scatter Op ( #3400 )
...
This PR adds OnnxToTorch support for Scatter op
2024-05-31 07:36:48 +00:00
afca88a058
[NFC] Change to *cast instead of .*cast variants ( #3405 )
...
Member casts have been deprecated. Changing over a bunch of the member
cast calls to the global templated variants to remove deprecation
warnings.
2024-05-30 23:45:13 -07:00
074098d20c
Modifies onnx resize lowering to fix numerical issues ( #3381 )
...
Updates:
- some unsupported modes are now going to report a match failure for
unsupported coordinate transformation modes.
- fixes a bug that was introduced in the last patch for resize (my
bad...)
- uses actual x and y coordinates for computing weights in bilinear
interpolation (rather than eps modified values)
- slightly simplifies the bilinear interpolation payload for readability
and performance
- passes coordinate transformation mode information from an onnx.Resize
op to the mode string for the aten._interpolate op. This allows us to
perform custom logic in the torch->linalg lowering to support
onnx.Resize options without losing the default behaviors of the
interpolate op.
2024-05-30 20:34:37 -04:00
27169dcda9
Replace some depreciated uses of cast ( #3343 )
...
Contributing towards #3299
2024-05-23 09:01:47 -07:00
fcf48872b3
[ONNX] Implement Softsign op ( #3373 )
2024-05-21 12:10:26 -07:00
297c270980
onnx.Resize and aten._interpolate : allow n spatial dims. ( #3368 )
...
The old lowering only had logic for 2d (i.e. images). this patch allows
interpolation for n spatial dims, which is required for some 3d vision
models such as
- onnx/models/pytorch-3dunet_vaiq_int8
which successfully compiles and runs with this patch.
2024-05-20 13:35:27 -07:00
513d89c16d
Add support for the onnx.SequenceLength op. ( #3362 )
2024-05-17 12:17:43 -07:00
72e38dcbbc
Add support for the onnx.SequenceConstruct op. ( #3316 )
2024-05-17 22:51:28 +05:30
73b3065a94
[ONNX] Reduces Transpose Opset Version ( #3302 )
...
As mentioned in issue #3290 , the difference between onnx.Transpose in
versions 1 and 13 is minimal, and therefore should be supported with the
same conversion pattern.
2024-05-14 21:38:56 +05:30
20f312853c
[MLIR][ONNX] Add OnnxToTorch support for ReduceLogSumExp Op ( #3201 )
...
This commit adds the OnnxToTorch support for ReduceLogSumExp op
2024-05-14 09:54: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
17c3c15131
[ONNX] Add OnnxToTorch lowering for SoftmaxCrossEntropyLoss op ( #3278 )
...
Signed-Off By: Vivek Khandelwal <vivekkhandelwal1424@gmail.com>
2024-05-06 17:26:32 +05:30
321b844df7
Revert hyperbolic trigonometric decompositions ( #3271 )
...
We should be using the `torch` path and handling decomposition in the
`math` dialect.
2024-05-03 12:06:44 -04:00
a46fe2c9db
[MLIR][ONNX] Add OnnxToTorch support for ReduceSumSquare Op ( #3188 )
...
This commit adds the OnnxToTorch support for ReduceSumSquare ops.
---------
Co-authored-by: Ubuntu <archana@archana-cpu.judsoscro3wupi0qm4bjlj5m3b.bx.internal.cloudapp.net>
2024-05-02 22:17:45 +05:30
0bb62e4347
Revert Onnx.Selu lowering to corresponding Aten op ( #3275 )
2024-05-02 09:00:24 -07:00
05f8b69bf6
[MLIR][TORCH] Add OnnxToTorch support for BlackmanWindow function ( #3181 )
...
Implements OnnxToTorch lowering for the BlackmanWindow Function.
2024-04-30 12:21:27 -04:00
b64c22cfc1
Fix onnx sinh lowering ( #3253 )
...
iree tests `test_sinh` and `test_sinh_example` passed
2024-04-30 00:44:41 -07:00
b1e2241479
[ONNX] Fix Onnx.Selu lowering and canonicalizer for IntImplicit op ( #3221 )
...
Signed-Off By: Vivek Khandelwal <vivekkhandelwal1424@gmail.com>
2024-04-29 04:00:01 +00:00
6679728c56
Fix deprecated uses of cast/dyn_cast/dyn_cast_or_null/isa ( #3243 )
...
Like #3130 , gradually replace the deprecated code
https://github.com/llvm/mlir-www/blob/main/website/content/deprecation/_index.md#deprecated
2024-04-27 14:00:56 -07:00
ac11ec796d
[MLIR][ONNX] Add OnnxToTorch support for ReduceLogSum Op ( #3229 )
...
This commit adds the OnnxToTorch support for ReduceLogSum op
2024-04-25 19:37:57 -04:00
Samu Tamminen
jinchen
giacs-epic
Rob Suderman
aldesilv
Marius Brehler
Rob Suderman
Gaurav Shukla
Vinayak Dev
pdhirajkumarprasad
Xida Ren (Cedar)
zjgarvey
Vivek Khandelwal
Suraj Sudhir
Matthias Gehre
Surya Jasper
RattataKing
Andrew Woloszyn
Archana Ramalingam
penguin_wwy