mlir-hlo/tests
Abhishek Varma da6593e960 PR #50073: [MLIR] Add GatherOp lowering from lmhlo to Affine.
Imported from GitHub PR https://github.com/tensorflow/tensorflow/pull/50073

-- Lowering of `GatherOp` is added from lmhlo to Affine. The lowering
   has been added as a part of `lhlo-legalize-to-affine` pass.

Signed-off-by: Abhishek Varma <abhishek.varma@polymagelabs.com>
Copybara import of the project:

--
5b3dcd4ab31a69f305cd079b869ee35ba6dc8bf5 by Abhishek Varma <abhishek.varma@polymagelabs.com>:

[MLIR] Add GatherOp lowering from lmhlo to Affine.

-- Lowering of `GatherOp` is added from lmhlo to Affine. The lowering
   has been added as a part of `lhlo-legalize-to-affine` pass.

Signed-off-by: Abhishek Varma <abhishek.varma@polymagelabs.com>
PiperOrigin-RevId: 380052157
2021-06-17 14:55:49 -07:00
..
end2end Integrate LLVM at llvm/llvm-project@8396aeb07c 2021-03-31 08:01:34 -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
broadcast_propagation.mlir [MLIR][HLO] Extend broadcast propagation pass to enable more fusion 2021-06-09 03:03:37 -07:00
canonicalize.mlir Handle empty tensors in SimplifyConcatSlice. 2021-06-09 02:15:47 -07:00
chlo_infer_shape_type_methods.mlir Integrate LLVM at llvm/llvm-project@671f0e2e18 2021-04-28 16:37:53 -07:00
chlo_legalize_to_hlo_broadcasts.mlir Integrate LLVM at llvm/llvm-project@cb65419b1a 2021-05-26 04:47:24 -07:00
chlo_legalize_to_mhlo.mlir Fix Cosh approximation for F16. 2021-06-10 06:16:44 -07:00
chlo_ops.mlir [MLIR][HLO] Implement `RegionBranchOpInterface` for rank specialization cluster 2021-05-11 09:03:05 -07:00
concatenate.mlir Rename `xla_hlo` dialect to `mhlo` 2020-07-30 22:32:50 +00:00
convert.mlir [MHLO] Don't crash trying to constant fold mhlo.convert on complex 2021-05-11 05:15:57 -07:00
disc_ral_inject_execution_context.mlir PR #50191: [MLIR][DISC] Add RAL (Runtime abstraction layer) Dialect 2021-06-14 11:27:43 -07: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-only-dynamic.mlir PR #50271: [MLIR][DISC] Bufferize GatherOp and DynamicGatherOp 2021-06-16 13:47:56 -07:00
hlo-legalize-to-lhlo-unranked.mlir PR #49454: [MLIR][DISC] Upgrade to use the new `reifyReturnTypeShapes` interface. 2021-05-24 10:11:55 -07:00
hlo-legalize-to-lhlo.mlir PR #49454: [MLIR][DISC] Upgrade to use the new `reifyReturnTypeShapes` interface. 2021-05-24 10:11:55 -07:00
hlo-legalize-to-linalg.mlir Add tests for lowering HLO_ExpOp for complex types to Linalg. 2021-06-17 06:34:19 -07:00
inlining.mlir Rename `xla_hlo` dialect to `mhlo` 2020-07-30 22:32:50 +00:00
legalize-control-flow.mlir Restrict MHLO control flow ops to single-block regions 2021-03-30 17:51:03 -07:00
legalize-to-std.mlir Integrate LLVM at llvm/llvm-project@b24436ac96 2021-03-23 12:20:17 -07:00
legalize-trigonometric-to-approximation.mlir Fix tanh lowering for NaN input. 2021-03-24 06:34:36 -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@b24436ac96 2021-03-23 12:20:17 -07:00
lhlo-fusion.mlir PR #50020: [MLIR][DISC] support fusion on buffer 2021-06-16 09:51:29 -07:00
lhlo-legalize-select-and-scatter.mlir Integrate LLVM at llvm/llvm-project@4b13b7581d 2021-04-27 12:19:05 -07:00
lhlo-legalize-tensor-load-op.mlir PR #49598: [MLIR][DISC] legalize tensor_load inserted during hlo-to-lhlo conversion 2021-06-01 16:27:56 -07:00
lhlo-legalize-to-affine.mlir PR #50073: [MLIR] Add GatherOp lowering from lmhlo to Affine. 2021-06-17 14:55:49 -07: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 Add tests for lowering HLO_ExpOp for complex types to Linalg. 2021-06-17 06:34:19 -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 [HLO] Adopt custom syntax for convolution dims and window attributes for LMHLO_GPU 2021-05-20 09:41:48 -07:00
lhlo_ops.mlir [HLO] Add AllReduceScatter to MHLO and LMHLO dialects. 2021-06-14 09:37:07 -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 Generate Equal and NotEqual kernels for complex types. 2021-04-15 00:35:52 -07: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
ops.mlir [HLO] Add AllReduceScatter to MHLO and LMHLO dialects. 2021-06-14 09:37:07 -07:00
optimize-hlo.mlir Add an optimization that converts some Gathers to Slices. 2020-07-30 22:34:10 +00:00
rank-specialization.mlir [MLIR][KernelGen] Better rank specialization for clusters containing `mhlo.select` 2021-06-17 04:06:08 -07:00
reduce.mlir Integrate LLVM at llvm/llvm-project@cb65419b1a 2021-05-26 04:47:24 -07: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