mlir-hlo/tests
Hanhan Wang 2e0ee7759b Add support for lowering mhlo.torch_index_select to Linalg on tensors.
The change upstreams the pattern from IREE repo to MHLO repo.

PiperOrigin-RevId: 363406294
2021-03-17 06:33:41 -07:00
..
end2end Integrate LLVM at llvm/llvm-project@678241795c 2021-03-16 13:33:00 -07:00
CMakeLists.txt Add CMake files and lit configurations, enough for `ninja check-mlir-hlo` to pass on all the tests 2020-08-07 22:14:34 -07:00
canonicalize.mlir Integrate LLVM at llvm/llvm-project@678241795c 2021-03-16 13:33:00 -07:00
chlo_infer_shape_type_methods.mlir [MLIR][KernelGen] Implement InferShapedTypeOpInterface for `mhlo.compare/select` 2020-09-17 07:10:10 -07:00
chlo_legalize_to_hlo_broadcasts.mlir [mlir][hlo] Make select ready for dynamic shapes (ranked only for now) 2021-02-18 08:08:40 -08:00
chlo_legalize_to_mhlo.mlir [MLIR][KernelGen] Fix zeta lowering at poles 2021-03-10 01:09:10 -08:00
chlo_ops.mlir Add MinimumBroadcastShapesOp to chlo dialect. 2021-03-01 02:23:52 -08:00
concatenate.mlir Rename `xla_hlo` dialect to `mhlo` 2020-07-30 22:32:50 +00:00
convert.mlir Fix constant folding of mhlo.convert op with i1 element types 2020-11-23 14:18:28 -08:00
hlo-legalize-gather-to-torch-index-select.mlir Add a transform for Gathers to torch_index_select. 2020-07-30 22:34:32 +00:00
hlo-legalize-to-lhlo-unranked.mlir Integrate LLVM at llvm/llvm-project@678241795c 2021-03-16 13:33:00 -07:00
hlo-legalize-to-lhlo.mlir Integrate LLVM at llvm/llvm-project@678241795c 2021-03-16 13:33:00 -07:00
hlo-legalize-to-linalg.mlir Add support for lowering mhlo.torch_index_select to Linalg on tensors. 2021-03-17 06:33:41 -07:00
hlo-transform-unranked.mlir Remove rank 1 specialization from TransformUnrankedHloPass. 2021-03-04 05:24:53 -08:00
inlining.mlir Rename `xla_hlo` dialect to `mhlo` 2020-07-30 22:32:50 +00:00
legalize-control-flow.mlir Integrate LLVM at llvm/llvm-project@0cf7e4b252 2020-12-16 20:30:17 -08:00
legalize-to-std.mlir Integrate LLVM at llvm/llvm-project@e2d7d3cb0e 2021-01-14 18:05:33 -08:00
legalize-trigonometric-to-approximation.mlir [MLIR][MHLO] Approximate `tf.Tanh` as constant +/-1 for small/large values 2021-03-16 10:14:30 -07:00
legalize_to_scf.mlir Integrate LLVM at llvm/llvm-project@0cf7e4b252 2020-12-16 20:30:17 -08:00
lhlo-fuse-linalg.mlir Integrate LLVM at llvm/llvm-project@678241795c 2021-03-16 13:33:00 -07:00
lhlo-legalize-select-and-scatter.mlir Integrate LLVM at llvm/llvm-project@678241795c 2021-03-16 13:33:00 -07:00
lhlo-legalize-to-affine.mlir [mlir][hlo] Make min/max always propagate NaNs 2021-01-25 09:04:16 -08:00
lhlo-legalize-to-gpu.mlir Integrate LLVM at llvm/llvm-project@678241795c 2021-03-16 13:33:00 -07:00
lhlo-legalize-to-linalg.mlir Integrate LLVM at llvm/llvm-project@678241795c 2021-03-16 13:33:00 -07:00
lhlo-legalize-to-parallel-loops.mlir Integrate LLVM at llvm/llvm-project@678241795c 2021-03-16 13:33:00 -07:00
lhlo_gpu_ops.mlir Integrate LLVM at llvm/llvm-project@678241795c 2021-03-16 13:33:00 -07:00
lhlo_ops.mlir Integrate LLVM at llvm/llvm-project@678241795c 2021-03-16 13:33:00 -07:00
lit.cfg.py Add CMake files and lit configurations, enough for `ninja check-mlir-hlo` to pass on all the tests 2020-08-07 22:14:34 -07:00
lit.site.cfg.py.in Add license header to lit.site.cfg.py.in 2020-08-07 22:22:24 -07:00
lower-complex.mlir Removed Op(Complex, Real) lowering to address complex type inference issue 2020-11-06 15:24:23 -08:00
lower-general-dot.mlir Cleanup build rule names in compiler/mlir/hlo to remove the redundant/obsolete xla_ prefix 2020-07-30 22:33:29 +00:00
materialize-broadcasts.mlir Cleanup build rule names in compiler/mlir/hlo to remove the redundant/obsolete xla_ prefix 2020-07-30 22:33:29 +00:00
mhlo-fusion.mlir Cleanup build rule names in compiler/mlir/hlo to remove the redundant/obsolete xla_ prefix 2020-07-30 22:33:29 +00:00
mhlo_infer_shape_type_methods.mlir [HLO:MLIR] Make binary op type reification emit shape_of instead of tensor ops 2021-03-11 02:01:35 -08:00
move_up_dynamic_broadcasts_for_fusion.mlir [MLIR][MHLO] Allow recursion in the shape_of mover 2021-03-11 02:52:21 -08:00
ops.mlir Avoid creating tuple type only for verification 2021-03-15 17:58:19 -07:00
optimize-hlo.mlir Add an optimization that converts some Gathers to Slices. 2020-07-30 22:34:10 +00:00
reduce.mlir Rename `xla_hlo` dialect to `mhlo` 2020-07-30 22:32:50 +00:00
reshape.mlir Rename `xla_hlo` dialect to `mhlo` 2020-07-30 22:32:50 +00:00
reverse.mlir Rename `xla_hlo` dialect to `mhlo` 2020-07-30 22:32:50 +00:00
sink-constants-to-control-flow.mlir Sink standard dialect constants in sink_constants_to_control_flow pass 2020-08-03 19:30:29 -07:00
transpose.mlir Rename `xla_hlo` dialect to `mhlo` 2020-07-30 22:32:50 +00:00
tuple.mlir Rename `xla_hlo` dialect to `mhlo` 2020-07-30 22:32:50 +00:00
unfuse_batch_norm.mlir Integrate LLVM at llvm/llvm-project@678241795c 2021-03-16 13:33:00 -07:00