colin
/
mlir-hlo
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
mlir-hlo/tests
History
Adrian Kuegel 4033a56750 Add special cases for SelectOp rank specialization. 
We now use the same special cases for all ops with arity >= 2.
For binary ops, we now have only one special case if at least one of the
operands has exactly one element. In that case, we reshape both operands to
rank 1. Before, we had separate special cases whether the left-hand side
or the right-hand side have a scalar shape.

PiperOrigin-RevId: 366005835
 		2021-03-31 04:28:51 -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 Canonicalize MHLO Case and If Ops with constant conditions 2021-03-30 17:58:01 -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 [MHLO:linalg] Lower all dynamic broadcasts of static shapes to linalg.generic 2021-03-19 03:52:02 -07:00
hlo-transform-unranked.mlir Add special cases for SelectOp rank specialization. 2021-03-31 04:28:51 -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-legalize-select-and-scatter.mlir Integrate LLVM at llvm/llvm-project@5657f93e78 2021-03-23 06:15:46 -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] Do not yield results of ops that were moved out of assuming regions 2021-03-31 03:50:27 -07:00
ops.mlir Verify mhlo.if region return types match op 2021-03-30 17:57:20 -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