colin
/
mlir-hlo
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
mlir-hlo/tests
History
Adrian Kuegel e6a1f5f0f9 Add MinimumBroadcastShapesOp to chlo dialect. 
This op is useful for rank specialization of broadcasts. Kernel Generator
needs to generate one kernel for each rank, so if we can minimize the rank
of the broadcast shape, we can support more cases with the same number of
special-cased kernels.

PiperOrigin-RevId: 360137827
 		2021-03-01 02:23:52 -08:00
..
end2end Integrate LLVM at llvm/llvm-project@2bfe27da17 2021-02-12 08:32:03 -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 Add MLIR generated kernel for Angle kernel. 2021-02-02 00:47:20 -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 [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][CHLO] Add legalization for `chlo.polygamma` to MHLO 2021-02-17 08:33:01 -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 [KERNEL_GEN] Add a pattern to bufferize `mhlo.reshape(<unranked_tensor>)`. 2021-02-10 06:32:21 -08:00
hlo-legalize-to-lhlo.mlir Lower Expm1 kernel to math.ExpM1. 2021-02-18 04:54:23 -08:00
hlo-legalize-to-linalg.mlir Fix the shape of linalg.init_tensor in conv op lowering. 2021-02-26 09:34:11 -08:00
hlo-transform-unranked.mlir [MLIR][HLO] Remove duplicate `PopulateTransformUnrankedHloPatterns` 2021-02-23 07:50:47 -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@2bfe27da17 2021-02-12 08:32:03 -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@2bfe27da17 2021-02-12 08:32:03 -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@91e7a17133 2021-02-04 13:42:31 -08:00
lhlo-legalize-to-linalg.mlir [mhlo] Lower int->bool to a comparison with zero 2021-02-15 06:38:09 -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:LHLO] Add optional call target arg mapping to LMHLO CustomCall operations. 2021-02-22 08:43:00 -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 Support dynamic-shaped operand in verification of BroadcastInDim. 2021-02-17 16:18:09 -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