Tres Popp
ba2ee556f1
Handle negative exponents for lowering of hlo.pow
...
PiperOrigin-RevId: 352382812
2021-01-18 03:47:28 -08:00
A. Unique TensorFlower
bcdb3c3548
[MLIR] Lower mhlo.clamp to linalg
...
PiperOrigin-RevId: 351998800
2021-01-15 06:45:38 -08:00
A. Unique TensorFlower
9a1abaa212
Integrate LLVM at llvm/llvm-project@e2d7d3cb0e
...
Updates LLVM usage to match
[e2d7d3cb0ead](https://github.com/llvm/llvm-project/commit/e2d7d3cb0ead )
PiperOrigin-RevId: 351915841
2021-01-14 18:05:33 -08:00
Hanhan Wang
300a7c11ce
Upstream mhlo.dot_general lowering to Linalg to MHLO repo
...
PiperOrigin-RevId: 351514250
2021-01-12 22:08:46 -08:00
Hanhan Wang
8f58f844e5
Upstream mhlo.dot lowering to Linalg to MHLO repo.
...
We prototyped the lowering from mhlo.dot to linalg.matmul in IREE. Since Linalg
now supports matmul in tensors world, we can move the lowering logic to tensors
world, and upstream to legalize_to_linalg.cc. The patch lowers the mhlo.dot to
the linalg.matmul/matvec/dot in tensors world.
PiperOrigin-RevId: 351184911
2021-01-11 10:35:24 -08:00
Alexander Belyaev
180f917446
[KERNEL_GEN] Add a pattern for hlo.dyn_broadcast->linalg to enable is_inf kernel.
...
PiperOrigin-RevId: 351179620
2021-01-11 10:13:31 -08:00
A. Unique TensorFlower
b0bf2ef45b
Integrate LLVM at llvm/llvm-project@c3acda0798
...
Updates LLVM usage to match
[c3acda0798f9](https://github.com/llvm/llvm-project/commit/c3acda0798f9 )
PiperOrigin-RevId: 348896724
2020-12-23 23:53:54 -08:00
Tres Popp
a42213b870
Define lowering of [l]mhlo.pow.
...
For floating point operations, this uses std.pow.
For integer operations, this lowers to a loop.
This adds a dependency on scf.
PiperOrigin-RevId: 348537232
2020-12-21 15:27:40 -08:00
Alexander Belyaev
65222893ae
[KERNEL_GEN] Convert LHLO AddOp, SubOp (ComplexType) to complex ops.
...
PiperOrigin-RevId: 347805898
2020-12-16 05:45:06 -08:00
Benjamin Kramer
9930c20c31
[mlir][hlo] Fix lowering of NE comparison. It should return true if either side is NaN
...
PiperOrigin-RevId: 346988987
2020-12-11 06:46:14 -08:00
Alexander Belyaev
c36afd275e
[HLO] Add a pattern for HLO ConstOp to HLO -> Linalg conversion.
...
PiperOrigin-RevId: 346718273
2020-12-09 23:24:57 -08:00
Stephan Herhut
c3790af758
Add plumbing for or and xor to hlo to lhlo and linalg lowerings.
...
PiperOrigin-RevId: 346311314
2020-12-08 06:39:02 -08:00
Benjamin Kramer
5235eceea0
Lower mhlo shifts to linalg
...
PiperOrigin-RevId: 346161253
2020-12-07 13:02:32 -08:00
A. Unique TensorFlower
e866aac3ac
Integrate LLVM at llvm/llvm-project@72ddd559b8
...
Updates LLVM usage to match
[72ddd559b8aa](https://github.com/llvm/llvm-project/commit/72ddd559b8aa )
PiperOrigin-RevId: 340292055
2020-11-02 12:29:52 -08:00
Ahmed S. Taei
39389587d2
{mhlo.is_finite, lmhlo.is_finite} -> linalg.generic conversion
...
PiperOrigin-RevId: 334414295
2020-09-29 10:50:42 -07:00
Hanhan Wang
b29dd5ef8f
Fix a bug in the case check of reshape op lowering.
...
PiperOrigin-RevId: 332044191
2020-09-16 11:06:16 -07:00
Hanhan Wang
cf1b2b2d84
Add a test for lowering mhlo.floor to Linalg on tensors.
...
PiperOrigin-RevId: 331514657
2020-09-14 03:30:43 -07:00
Hanhan Wang
bfd629ecb0
Enhance lowering reshape op to Linalg.
...
Handle non-expansion and non-collapsion cases by rewriting it to two reshape
ops.
PiperOrigin-RevId: 327926863
2020-08-21 23:27:34 -07:00
Benjamin Kramer
b7c4314e7f
Allow running tests with FILECHECK_OPTS=-enable-var-scope
...
PiperOrigin-RevId: 323420636
2020-08-03 19:27:38 -07:00
Hanhan Wang
8f262ae8f5
Add support for lowering mhlo.iota to Linalg.
...
PiperOrigin-RevId: 322799853
2020-07-30 22:34:45 +00:00
Mehdi Amini
8900222fed
Rename `xla_hlo` dialect to `mhlo`
...
This is part of the current refactoring of the HLO related dialect.
`xla_hlo` will be reintroduced in a new form later.
PiperOrigin-RevId: 319916753
2020-07-30 22:32:50 +00:00
Mehdi Amini
fa057cc0bc
Move HLO tests to `third_party/tensorflow/compiler/mlir/hlo/`
...
Also add a localized `mlir-hlo-opt` binary for the testing of
tensorflow/compiler/mlir/hlo/... ; this directory is intended to be self-contained
and depend only on MLIR.
PiperOrigin-RevId: 319878984
2020-07-30 22:32:45 +00:00