Commit Graph

1069 Commits

Author SHA1 Message Date
Jacques Pienaar ab4c03b84b Replace operands now injected in OpBuilder
These are now injected automatically.

PiperOrigin-RevId: 336356284
2020-10-09 13:39:33 -07:00
Tres Popp f6af1fc134 Support hlo to lhlo buffer placement through shape.assuming ops.
PiperOrigin-RevId: 336287728
2020-10-09 07:14:12 -07:00
Stephan Herhut d986bd7ad7 Use tensor_cast instead of mhlo::reshape in the lowering of unranked binary operations.
We know that the value already is a scalar and we just want to update the type, so no need to reshape anything.

PiperOrigin-RevId: 336252315
2020-10-09 01:46:48 -07:00
Rahul Joshi 41436ea0d9 [NFC] Extract common attributes of MHLO and LMHLO ConvOp
- Create a common class that hold ConvOp attributes that are common to MHLO and
  LHLO Dialects and use Tablegen DAG concat to append these common attributes to
  dialect specific arguments in ODS for ConvOp.

PiperOrigin-RevId: 336114003
2020-10-08 10:29:32 -07:00
A. Unique TensorFlower 3a99598b59 Integrate LLVM at llvm/llvm-project@c102488293
Updates LLVM usage to match
[c10248829357](https://github.com/llvm/llvm-project/commit/c10248829357)

PiperOrigin-RevId: 336090676
2020-10-08 08:32:44 -07:00
A. Unique TensorFlower 4c3072c098 Integrate LLVM at llvm/llvm-project@9908ee5670
Updates LLVM usage to match
[9908ee567059](https://github.com/llvm/llvm-project/commit/9908ee567059)

PiperOrigin-RevId: 335944587
2020-10-07 13:47:02 -07:00
Pavel Krajcevski ac483f4091 Remove superfluous quote from lhlo_ops.td
PiperOrigin-RevId: 335921873
2020-10-07 12:06:36 -07:00
A. Unique TensorFlower 673588c864 Integrate LLVM at llvm/llvm-project@5e4409f308
Updates LLVM usage to match
[5e4409f30817](https://github.com/llvm/llvm-project/commit/5e4409f30817)

PiperOrigin-RevId: 335852101
2020-10-07 06:15:47 -07:00
A. Unique TensorFlower 3736c5542f [MLIR][KernelGen] Fix unranked codegeneration in kernel generator
PiperOrigin-RevId: 335847086
2020-10-07 05:39:55 -07:00
A. Unique TensorFlower 1f14527609 Integrate LLVM at llvm/llvm-project@b45b5166f8
Updates LLVM usage to match
[b45b5166f8f9](https://github.com/llvm/llvm-project/commit/b45b5166f8f9)

PiperOrigin-RevId: 335836653
2020-10-07 04:18:12 -07:00
A. Unique TensorFlower af88c2999b Integrate LLVM at llvm/llvm-project@3cb8347c94
Updates LLVM usage to match
[3cb8347c94a0](https://github.com/llvm/llvm-project/commit/3cb8347c94a0)

PiperOrigin-RevId: 335767863
2020-10-06 18:44:02 -07:00
A. Unique TensorFlower 10a4e1ad2d Integrate LLVM at llvm/llvm-project@a48d480e1f
Updates LLVM usage to match
[a48d480e1f7e](https://github.com/llvm/llvm-project/commit/a48d480e1f7e)

PiperOrigin-RevId: 335597620
2020-10-06 02:19:35 -07:00
Lucy Fox 84277ed784 Remove unused `using` statements.
PiperOrigin-RevId: 335517711
2020-10-05 15:53:59 -07:00
A. Unique TensorFlower ab1a2572d4 Integrate LLVM at llvm/llvm-project@3641d375f6
Updates LLVM usage to match
[3641d375f674](https://github.com/llvm/llvm-project/commit/3641d375f674)

PiperOrigin-RevId: 335502425
2020-10-05 14:41:46 -07:00
A. Unique TensorFlower a33441907c [MLIR][KernelGen] Add E2E test for `tf.Acos`
PiperOrigin-RevId: 335419310
2020-10-05 08:14:36 -07:00
A. Unique TensorFlower bae0815ef0 [MLIR][KernelGen] Legalize `atan` to approximation
PiperOrigin-RevId: 335417836
2020-10-05 08:05:52 -07:00
A. Unique TensorFlower 7f84a86cf5 [MLIR][KerneGen] Lower `tf.Atan` all the way to LLVM
PiperOrigin-RevId: 335394668
2020-10-05 05:07:13 -07:00
A. Unique TensorFlower c9f4755863 Integrate LLVM at llvm/llvm-project@22664a3251
Updates LLVM usage to match
[22664a325167](https://github.com/llvm/llvm-project/commit/22664a325167)

PiperOrigin-RevId: 335391717
2020-10-05 04:42:21 -07:00
Alexander Belyaev d927e32451 [HLO] Clean-up dynamic allocation in hlo-legalize-to-lhlo pass.
PiperOrigin-RevId: 335385243
2020-10-05 03:55:24 -07:00
A. Unique TensorFlower 7367eac074 Add folder for mhlo::remainder
PiperOrigin-RevId: 335372628
2020-10-05 02:20:01 -07:00
A. Unique TensorFlower c9e249c124 Integrate LLVM at llvm/llvm-project@88c9162c9d
Updates LLVM usage to match
[88c9162c9d47](https://github.com/llvm/llvm-project/commit/88c9162c9d47)

PiperOrigin-RevId: 335144150
2020-10-02 19:32:46 -07:00
A. Unique TensorFlower 5146c29cbf Integrate LLVM at llvm/llvm-project@0f0cbcc4b1
Updates LLVM usage to match
[0f0cbcc4b166](https://github.com/llvm/llvm-project/commit/0f0cbcc4b166)

PiperOrigin-RevId: 335119390
2020-10-02 16:08:17 -07:00
Tim Shen c708bfd6d0 [MLIR] Add cbrt, reduce-precision, and bitcast ops to MHLO.
PiperOrigin-RevId: 335109804
2020-10-02 15:13:18 -07:00
Tres Popp bcf6fbf612 Integrate LLVM at llvm/llvm-project@8825fec37e
Updates LLVM usage to match
[8825fec37e73](https://github.com/llvm/llvm-project/commit/8825fec37e73)

Adapt to upstream subview / subtensor changes.

PiperOrigin-RevId: 335023452
2020-10-02 07:34:47 -07:00
Tres Popp a3a8c498e1 Integrate LLVM at llvm/llvm-project@bfd7ee92cc
Updates LLVM usage to match
[bfd7ee92ccec](https://github.com/llvm/llvm-project/commit/bfd7ee92ccec)

PiperOrigin-RevId: 335003324
2020-10-02 04:39:31 -07:00
Adrian Kuegel 3eb767b43d Add missing lowering step for IsFiniteOp.
Also add a BUILD target for generating the GPU kernel.

PiperOrigin-RevId: 334993362
2020-10-02 03:09:00 -07:00
Chao Xie 5f303440da [MLIR][KerneGen] Lower `tf.Atan` all the way to LLVM
PiperOrigin-RevId: 334843070
2020-10-01 10:25:51 -07:00
Tres Popp aa65e49ad2 Integrate LLVM at llvm/llvm-project@5101e7e8dd
Updates LLVM usage to match
[5101e7e8dd01](https://github.com/llvm/llvm-project/commit/5101e7e8dd01)

PiperOrigin-RevId: 334812211
2020-10-01 07:49:58 -07:00
A. Unique TensorFlower 458e861254 [MLIR][KerneGen] Lower `tf.Atan` all the way to LLVM
PiperOrigin-RevId: 334810730
2020-10-01 07:38:40 -07:00
Tres Popp 73f461080c Integrate LLVM at llvm/llvm-project@fcf70e1e3b
Updates LLVM usage to match
[fcf70e1e3b1d](https://github.com/llvm/llvm-project/commit/fcf70e1e3b1d)

PiperOrigin-RevId: 334796954
2020-10-01 05:58:50 -07:00
A. Unique TensorFlower 049ca060a1 [MLIR][KernelGen] Legalize `atan2` to approximation
Legalize `atan2` analogously to XLA.  `atan2` is first reduced to `atan` on the
interval [-1, 1] and subsequently approximated.  This CL also adds e2e tests for
trigonometric approximations.

PiperOrigin-RevId: 334794336
2020-10-01 05:34:48 -07:00
A. Unique TensorFlower 4b1809784a Support collapse_slice_dims in the mhlo.gather->mhlo.slice canonicalizer
PiperOrigin-RevId: 334774763
2020-10-01 02:46:49 -07:00
Tres Popp ae51900562 Integrate LLVM at llvm/llvm-project@ce5379f0f0
Updates LLVM usage to match
[ce5379f0f067](https://github.com/llvm/llvm-project/commit/ce5379f0f067)

PiperOrigin-RevId: 334774614
2020-10-01 02:45:28 -07:00
Rahul Joshi bce128b070 Introduce CustomCall operation in LHLO Dialect
- And add conversion from MHLO CustomCall to LHLO CustomCall
- According to XLA documentation, the called function should not be side effecting,
  so marking the argument MemRefs as MemRead.

PiperOrigin-RevId: 334737196
2020-09-30 20:56:18 -07:00
A. Unique TensorFlower e2ffba3f61 Integrate LLVM at llvm/llvm-project@700e63293e
Updates LLVM usage to match
[700e63293eea](https://github.com/llvm/llvm-project/commit/700e63293eea)

PiperOrigin-RevId: 334732681
2020-09-30 20:13:43 -07:00
Benjamin Kramer dfe64d3958 Implement InferShapedTypeOpInterface for mhlo.complex
Binary companion for 8bcd33e4b7

PiperOrigin-RevId: 334651523
2020-09-30 12:14:15 -07:00
Tres Popp 019c5ef106 Integrate LLVM at llvm/llvm-project@8c05c7c8d8
Updates LLVM usage to match
[8c05c7c8d87c](https://github.com/llvm/llvm-project/commit/8c05c7c8d87c)

PiperOrigin-RevId: 334598568
2020-09-30 07:57:41 -07:00
Benjamin Kramer c8919f8419 Implement InferShapedTypeOpInterface and use inferReturnTypes for mhlo.imag and mhlo.real
This makes the lhlo lowering work with dynamic shapes.

PiperOrigin-RevId: 334553472
2020-09-30 02:02:26 -07: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
Benjamin Kramer 336ee14538 Make lmhlo.sign to linalg lowering work for more floating point and integer types
PiperOrigin-RevId: 334400341
2020-09-29 09:52:13 -07:00
Tres Popp 04bf09382e Integrate LLVM at llvm/llvm-project@eb9f7c28e5
Updates LLVM usage to match
[eb9f7c28e5fe](https://github.com/llvm/llvm-project/commit/eb9f7c28e5fe)

PiperOrigin-RevId: 334388404
2020-09-29 08:54:26 -07:00
Benjamin Kramer 6459f12235 Lower mhlo.not to a xor with all ones
PiperOrigin-RevId: 334361499
2020-09-29 05:59:40 -07:00
Robert Suderman 26ac5baae4 Make mhlo.sort return variadic results instead of a tuple
Tuple is only used on XLA's sort to return multiple inputs. MLIR supports
multiple inputs, switch to a tuple return.

PiperOrigin-RevId: 334226937
2020-09-28 13:32:23 -07:00
A. Unique TensorFlower be2ffd2e21 Integrate LLVM at llvm/llvm-project@6caf3fb817
Updates LLVM usage to match
[6caf3fb81786](https://github.com/llvm/llvm-project/commit/6caf3fb81786)

PiperOrigin-RevId: 333820612
2020-09-25 15:28:22 -07:00
A. Unique TensorFlower eec7eb72fc Integrate LLVM at llvm/llvm-project@81a408808f
Updates LLVM usage to match
[81a408808f66](https://github.com/llvm/llvm-project/commit/81a408808f66)

PiperOrigin-RevId: 333600462
2020-09-24 14:37:27 -07:00
A. Unique TensorFlower 9125701a38 Integrate LLVM at llvm/llvm-project@ada1e2ffa1
Updates LLVM usage to match
[ada1e2ffa117](https://github.com/llvm/llvm-project/commit/ada1e2ffa117)

PiperOrigin-RevId: 333566305
2020-09-24 12:02:43 -07:00
Alexander Belyaev 4ddca0cd03 Integrate LLVM at llvm/llvm-project@a90d72127a
Updates LLVM usage to match
[a90d72127a81](https://github.com/llvm/llvm-project/commit/a90d72127a81)

PiperOrigin-RevId: 333488152
2020-09-24 04:27:08 -07:00
Alexander Belyaev 2e142a685c Integrate LLVM at llvm/llvm-project@beeceb92c0
Updates LLVM usage to match
[beeceb92c082](https://github.com/llvm/llvm-project/commit/beeceb92c082)

PiperOrigin-RevId: 333459746
2020-09-24 00:34:31 -07:00
Ahmed S. Taei 9c6640cbb6 Only apply GeneralDotOpLoweringPatterns for static shaped inputs
PiperOrigin-RevId: 333439680
2020-09-23 21:42:01 -07:00
A. Unique TensorFlower 34c6844dcc Integrate LLVM at llvm/llvm-project@af1d3e6559
Updates LLVM usage to match
[af1d3e655991](https://github.com/llvm/llvm-project/commit/af1d3e655991)

PiperOrigin-RevId: 333385949
2020-09-23 15:31:15 -07:00