mlir-hlo/lib/Dialect/mhlo/transforms
Benjamin Kramer a9cc1dcfa0 [mlir][hlo] Add basic rank-specialization for select
This just blows up everything to ranked (up to 6) and is probably quite slow.
This is sufficient to make kernelgen compile SelectV2.

PiperOrigin-RevId: 358777728
2021-02-22 02:41:12 -08:00
..
CMakeLists.txt Integrate LLVM at llvm/llvm-project@2bfe27da17 2021-02-12 08:32:03 -08:00
chlo_legalize_to_hlo.cc [mlir][hlo] Make select ready for dynamic shapes (ranked only for now) 2021-02-18 08:08:40 -08:00
chlo_legalize_to_hlo_pass.cc [MLIR][CHLO] Add legalization for `chlo.polygamma` to MHLO 2021-02-17 08:33:01 -08:00
chlo_legalize_to_hlo_patterns.td [MLIR][CHLO] Add `is_inf`, `is_pos_inf`, and `is_neg_inf` to CHLO dialect 2021-01-27 09:00:56 -08:00
hlo_legalize_to_lhlo.cc Lower Expm1 kernel to math.ExpM1. 2021-02-18 04:54:23 -08:00
legalize_control_flow.cc Fix MLIR include paths. 2020-12-17 00:56:04 -08:00
legalize_gather_to_torch_index_select.cc Integrate LLVM at llvm/llvm-project@d1978fa4bf 2021-02-05 07:42:06 -08:00
legalize_to_linalg.cc Lower Expm1 kernel to math.ExpM1. 2021-02-18 04:54:23 -08:00
legalize_to_standard.cc Integrate LLVM at llvm/llvm-project@d1978fa4bf 2021-02-05 07:42:06 -08:00
legalize_to_standard_patterns.td Lowering for mhlo.ceil to std.ceil 2020-08-12 16:15:35 -07:00
legalize_trigonometric_to_approximation.cc Integrate LLVM at llvm/llvm-project@2bfe27da17 2021-02-12 08:32:03 -08:00
lhlo_fuse_linalg.cc Integrate LLVM at llvm/llvm-project@d1978fa4bf 2021-02-05 07:42:06 -08:00
lhlo_legalize_to_affine.cc Integrate LLVM at llvm/llvm-project@d1978fa4bf 2021-02-05 07:42:06 -08:00
lhlo_legalize_to_gpu.cc Integrate LLVM at llvm/llvm-project@91e7a17133 2021-02-04 13:42:31 -08:00
lhlo_legalize_to_parallel_loops.cc [mlir][NFC] Replace usages or mlir/IR/StandardTypes.h with mlir/IR/BuiltinTypes.h 2020-12-15 00:59:29 -08:00
lower_complex.cc Integrate LLVM at llvm/llvm-project@d1978fa4bf 2021-02-05 07:42:06 -08:00
lower_complex_patterns.td Removed Op(Complex, Real) lowering to address complex type inference issue 2020-11-06 15:24:23 -08:00
lower_general_dot.cc Integrate LLVM at llvm/llvm-project@d1978fa4bf 2021-02-05 07:42:06 -08:00
materialize_broadcasts.cc More cleanup in mlir-hlo to prepare for the standalone build 2020-08-03 19:28:00 -07:00
materialize_broadcasts_pass.cc Integrate LLVM at llvm/llvm-project@0fc1aa22ee 2020-10-27 06:56:16 -07:00
mhlo_control_flow_to_scf.cc Fix MLIR include paths. 2020-12-17 00:56:04 -08:00
mhlo_fusion.cc More cleanup in mlir-hlo to prepare for the standalone build 2020-08-03 19:28:00 -07:00
optimize_mhlo.cc More cleanup in mlir-hlo to prepare for the standalone build 2020-08-03 19:28:00 -07:00
optimize_mhlo_pass.cc Integrate LLVM at llvm/llvm-project@d1978fa4bf 2021-02-05 07:42:06 -08:00
sink_constants_to_control_flow.cc Enable fallback legalization for MaxPoolGradGrad and MaxPool3DGradGrad ops 2020-11-13 13:49:05 -08:00
test_infer_shaped_type_pass.cc Integrate LLVM at llvm/llvm-project@d1978fa4bf 2021-02-05 07:42:06 -08:00
transform_unranked_hlo.cc [mlir][hlo] Add basic rank-specialization for select 2021-02-22 02:41:12 -08:00
unfuse_batch_norm.cc Integrate LLVM at llvm/llvm-project@96ef4f307d 2021-01-20 07:09:47 -08:00
unfuse_batch_norm_pass.cc Integrate LLVM at llvm/llvm-project@d1978fa4bf 2021-02-05 07:42:06 -08:00