torch-mlir/lib/Conversion/TorchToStablehlo
Stella Laurenzo 078d1e1a1d
Remove mlir-hlo (replace with stablehlo). (#2460)
We just have to do this: I ran into an issue today where I needed to make a one line patch to stablehlo to work around a compiler issue, and it is completely unapparent how to do so given that the mlir-hlo repo is a read-only export and is at the tail end of a multi-week integration chain from the open-source stablehlo repo.

We've discussed this often enough and gotten +1 from everyone that they are ok with taking the e2e testing hit if it becomes necessary: It is necessary as the current situation is unmanageable.

Looking at it, I expect it wouldn't actually be very difficult to build a little runner binary out of the stablehlo interpreter and subprocess call that in order to get the testing coverage back. I leave that as an exercise to the users of this part of the stack and recommend following the breadcrumbs from the deleted python/torch_mlir_e2e_test/stablehlo_backends/linalg_on_tensors.py file and the main.py changes.

Note that I am pointing us at a stablehlo fork for the moment until it is apparent that we don't need to carry any local patches to it. We can update this in a few days if everything is clear.
2023-09-12 19:10:02 -07:00
..
Basic.cpp Remove mlir-hlo (replace with stablehlo). (#2460) 2023-09-12 19:10:02 -07:00
CMakeLists.txt Remove mlir-hlo (replace with stablehlo). (#2460) 2023-09-12 19:10:02 -07:00
GatherScatter.cpp [MLIR][TORCH] Add StableHLO lowering for embedding_bag.padding_idx op 2023-09-05 21:32:23 +05:30
Linear.cpp [Stablehlo] fix promoteType() when input doesn't have DefiningOp (#2262) 2023-06-26 00:04:17 +08:00
Pooling.cpp [Stablehlo]Add support for AvgPool1dOp (#2268) 2023-07-25 14:09:53 +08:00
PopulatePatterns.h [StableHLO] Support for slice_scatter (#1960) 2023-03-22 13:41:04 -07:00
Reduction.cpp [MLIR][TORCH] add E2E support for aten.min op (#2422) 2023-08-29 12:12:41 -05:00
StablehloLegalizeUtils.cpp [Stablehlo] fix promoteType() when input doesn't have DefiningOp (#2262) 2023-06-26 00:04:17 +08:00
TorchToStablehlo.cpp [Stablehlo] support dynamic shape when convert aten.fill.Scalar (#2349) 2023-07-27 18:35:25 +08:00
ViewLike.cpp [MLIR][TORCH] Patch up Ops and their lowerings to deal with +ve `dim` 2023-04-14 13:12:56 +05:30