torch-mlir

Commit Graph

Author	SHA1	Message	Date
Yuanqiang Liu	8d5e2578b0	[Stablehlo] lowering aten.view to shape.num_elements + stablehlo.comp… (#3125 ) …ute_reshape_shape as that `aten.view` support at most one `-1` in dim list. The original calculation of `numel` is wrong when there is a `-1` in dim list.	2024-04-09 14:54:57 +08:00
Matthias Gehre	64d7626a52	Fixes for split tensor and slice (#2314 ) * RecomposeComplexOps: Remove dead slice op * lib/Dialect/Torch/IR/TorchOps.cpp: Fold slice ops even when they are on non-value tensors * lib/Conversion/TorchToTosa/TorchToTosa.cpp: Fix slice start/end out of range/none * lib/Dialect/Torch/IR/TorchOps.cpp: AtenSliceTensorOp::fold: Fold slices that go from 0:int_max * More tests for aten.split.Tensor	2023-07-20 09:53:54 +02:00
Yuanqiang Liu	c86f46bd70	[test] rename TorchToMhlo to TorchToStablehlo (#1995 )	2023-04-03 18:41:25 -07:00

Author

SHA1

Message

Date

Yuanqiang Liu

8d5e2578b0

[Stablehlo] lowering aten.view to shape.num_elements + stablehlo.comp… (#3125 )

…ute_reshape_shape

as that `aten.view` support at most one `-1` in dim list. The original
calculation of `numel` is wrong when there is a `-1` in dim list.

2024-04-09 14:54:57 +08:00

Matthias Gehre

64d7626a52

Fixes for split tensor and slice (#2314 )

* RecomposeComplexOps: Remove dead slice op

* lib/Dialect/Torch/IR/TorchOps.cpp: Fold slice ops even when they are on non-value tensors

* lib/Conversion/TorchToTosa/TorchToTosa.cpp: Fix slice start/end out of range/none

* lib/Dialect/Torch/IR/TorchOps.cpp: AtenSliceTensorOp::fold: Fold slices that go from 0:int_max

* More tests for aten.split.Tensor

2023-07-20 09:53:54 +02:00

Yuanqiang Liu

c86f46bd70

[test] rename TorchToMhlo to TorchToStablehlo (#1995 )

2023-04-03 18:41:25 -07:00

3 Commits (6abc7371c8279a678a757af2fbf085502bd7e696)