colin
/
mlir-hlo
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
mlir-hlo/tests
History
Lei Zhang 39589add22 Use the correct shape when converting mhlo.reshape 
If mhlo.reshape is not purely collapsing some consecutive operand
dimensions into result dimensions, we will generate two linalg
reshape op for it: the first one collapses all operand dimensions
into one dimension, and the second one expands it to all result
dimensions. For this case, the number of collapsed/expanded dimensions
should be coming strictly from the operand/result. It is different
from the case where we can generate one linalg reshape. For that case,
the reassociation map should have rank equal to the largest among
operand/result shape.

PiperOrigin-RevId: 354293826
 		2021-01-28 05:37:54 -08:00
..
end2end Integrate LLVM at llvm/llvm-project@96ef4f307d 2021-01-20 07:09:47 -08: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@96ef4f307d 2021-01-20 07:09:47 -08: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 Integrate LLVM at llvm/llvm-project@c3acda0798 2020-12-23 23:53:54 -08:00
chlo_legalize_to_mhlo.mlir [MLIR][CHLO] Add `chlo.lgamma` and lowering to `hlo` 2021-01-28 04:35:03 -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 [HLO] Delete LHLO memref cast ops and migrate to STD ones. 2020-11-04 09:26:34 -08:00
hlo-legalize-to-lhlo.mlir Integrate LLVM at llvm/llvm-project@96ef4f307d 2021-01-20 07:09:47 -08:00
hlo-legalize-to-linalg.mlir Use the correct shape when converting mhlo.reshape 2021-01-28 05:37:54 -08:00
hlo-transform-unranked.mlir [MLIR][CHLO] Add `is_inf`, `is_pos_inf`, and `is_neg_inf` to CHLO dialect 2021-01-27 09:00:56 -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 Integrate LLVM at llvm/llvm-project@e2d7d3cb0e 2021-01-14 18:05:33 -08: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@c3acda0798 2020-12-23 23:53:54 -08:00
lhlo-legalize-select-and-scatter.mlir Integrate LLVM at llvm/llvm-project@e2d7d3cb0e 2021-01-14 18:05:33 -08: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@72ddd559b8 2020-11-02 12:29:52 -08:00
lhlo-legalize-to-linalg.mlir Use the correct shape when converting mhlo.reshape 2021-01-28 05:37:54 -08:00
lhlo-legalize-to-parallel-loops.mlir Integrate LLVM at llvm/llvm-project@e2d7d3cb0e 2021-01-14 18:05:33 -08:00
lhlo_gpu_ops.mlir [XLA:GPU] Add layout attributes to LHLO_GPU Convolution operations. 2020-12-17 09:26:28 -08:00
lhlo_ops.mlir [MLIR:HLO] Extend AllReduce to support multiple inputs and results (to model tuples). 2021-01-26 17:25:22 -08: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 Integrate LLVM at llvm/llvm-project@96ef4f307d 2021-01-20 07:09:47 -08:00
ops.mlir [MLIR] Update tests to eliminate public function declarations. 2020-11-12 10:42:59 -08: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@96ef4f307d 2021-01-20 07:09:47 -08:00