mlir-hlo/lib/Dialect/mhlo/transforms
Tres Popp d086b8a0ec Correct HLO atan2 lowering in cases of -inf and -0 inputs.
This is being done by just removing the approximation and lowering to atan2 lib calls later to make the implementation the same as XLA. Note that if the approximation is brought back later, it can be fixed by changing the IR checking `less-than(X, 0)` to `less-than(copysign(X, 1), 0)`

PiperOrigin-RevId: 356253941
2021-02-08 06:58:04 -08:00
..
CMakeLists.txt Fix CMakeLists.txt 2021-01-21 12:42:35 -08:00
chlo_legalize_to_hlo.cc Implement lowering of chlo::zeta to mhlo dialect. 2021-02-03 07:50:05 -08:00
chlo_legalize_to_hlo_pass.cc Implement lowering of chlo::zeta to mhlo dialect. 2021-02-03 07:50:05 -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 Integrate LLVM at llvm/llvm-project@c85b6bf33c 2021-01-27 11:46:07 -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 Integrate LLVM at llvm/llvm-project@91e7a17133 2021-02-04 13:42:31 -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 Correct HLO atan2 lowering in cases of -inf and -0 inputs. 2021-02-08 06:58:04 -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 Add kernel definition for zeta operation. 2021-02-04 01:27:43 -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