mlir-hlo/tests
A. Unique TensorFlower 4ebcebf31c [MLIR][HLO] Exploit scalar properties in rank specialization lowering
Take advantage of the fact that scalars are already ranked and that they are
neutral elements to broadcasting. Do not reshape scalars, do not consider them
for broadcasting, and materialize ranked operations on scalars accordingly.

PiperOrigin-RevId: 375968371
2021-05-26 09:59:13 -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
canonicalize.mlir Integrate LLVM at llvm/llvm-project@cb65419b1a 2021-05-26 04:47:24 -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 [MLIR][KernelGen] Fix zeta lowering at poles 2021-03-10 01:09:10 -08: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
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 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 Support complex types when converting HLO multiply op. 2021-05-25 04:35:34 -07:00
hlo-transform-unranked.mlir Lower ReluGrad via chlo::BroadcastSelect. 2021-05-04 01:03:02 -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@4b13b7581d 2021-04-27 12:19:05 -07:00
lhlo-legalize-to-affine.mlir PR #47315: [MLIR] Add concatenateOp lowering from lmhlo to Affine. 2021-04-14 11:06:38 -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 Support complex types when converting HLO multiply op. 2021-05-25 04:35:34 -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 custom print/parse for window attributes of convolutions (in LMHLO) 2021-05-14 09:47:25 -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
move_up_dynamic_broadcasts_for_fusion.mlir [MLIR][HLO] Add `shape.broadcast` canonicalization to unblock broadcast moving 2021-05-05 07:16:49 -07:00
ops.mlir [HLO] Adopt custom syntax for convolution dimensions and window attributes (HLO) 2021-05-20 12:13:50 -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][HLO] Exploit scalar properties in rank specialization lowering 2021-05-26 09:59:13 -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