33c450e4cb
Fix the MHLO to LMHLO lowering of 'gather'
...
The lowering assumes that the 'gather' op attributes are identical in both MHLO and LMHLO. But that's not true; some time ago the MHLO version was changed to pack 4 of its attributes into a struct. By doing the same for the LMHLO version we both fix the lowering for this op and resolve a longstanding TODO.
PiperOrigin-RevId: 337943946
2020-10-19 15:14:05 -07:00
204cf7d544
Integrate LLVM at llvm/llvm-project@f402e682d0
...
Updates LLVM usage to match
[f402e682d0ef](https://github.com/llvm/llvm-project/commit/f402e682d0ef )
PiperOrigin-RevId: 337872985
2020-10-19 09:34:40 -07:00
7a983ea389
Add folder for mhlo::pad
...
PiperOrigin-RevId: 337827560
2020-10-19 04:21:44 -07:00
4a18aa41ee
Add folder to mhlo::round_nearest_afz
...
PiperOrigin-RevId: 337823786
2020-10-19 03:45:15 -07:00
92c531cf36
Integrate LLVM at llvm/llvm-project@b740899c50
...
Updates LLVM usage to match
[b740899c500b](https://github.com/llvm/llvm-project/commit/b740899c500b )
PiperOrigin-RevId: 337679767
2020-10-17 13:03:55 -07:00
27968619b7
Verify non-scalar inputs for HLO concat
...
XLA HLO concat does not accept scalars, so fail verification if this occurs. Avoids segfault when accessing an empty output shape.
PiperOrigin-RevId: 337618167
2020-10-16 19:39:31 -07:00
706718b4fb
Permit vector types in lmhlo to std lowering.
...
PiperOrigin-RevId: 337523303
2020-10-16 09:47:02 -07:00
2e30b59ddc
Extend hlo-fuse-linalg slightly to support aliased returns.
...
The fusion heuristic identifies the root of a fusion by checking whether an
output of a linalg operation is a function result. It did not consider outputs
flowing through aliasing operations (like casts).
PiperOrigin-RevId: 337479910
2020-10-16 04:04:23 -07:00
6a91ac6d8f
Integrate LLVM at llvm/llvm-project@51ff04567b
...
Updates LLVM usage to match
[51ff04567b2f](https://github.com/llvm/llvm-project/commit/51ff04567b2f )
PiperOrigin-RevId: 337477820
2020-10-16 03:40:23 -07:00
7ec0408776
Integrate LLVM at llvm/llvm-project@cfa7094e49
...
Updates LLVM usage to match
[cfa7094e49cf](https://github.com/llvm/llvm-project/commit/cfa7094e49cf )
PiperOrigin-RevId: 337464568
2020-10-16 01:31:55 -07:00
b56793297a
Integrate LLVM at llvm/llvm-project@f6bf2823c4
...
Updates LLVM usage to match
[f6bf2823c44e](https://github.com/llvm/llvm-project/commit/f6bf2823c44e )
PiperOrigin-RevId: 337425628
2020-10-15 18:44:35 -07:00
51cd4200b6
Make LMHLO's Dot have the same power as MHLO's DotGeneral.
...
PiperOrigin-RevId: 337391565
2020-10-15 15:09:06 -07:00
8fe5bc89bb
Integrate LLVM at llvm/llvm-project@50df5f24dc
...
Updates LLVM usage to match
[50df5f24dc33](https://github.com/llvm/llvm-project/commit/50df5f24dc33 )
PiperOrigin-RevId: 337292894
2020-10-15 06:08:13 -07:00
eb5a41da32
Integrate LLVM at llvm/llvm-project@196bee9648
...
Updates LLVM usage to match
[196bee9648a9](https://github.com/llvm/llvm-project/commit/196bee9648a9 )
PiperOrigin-RevId: 337280377
2020-10-15 04:12:11 -07:00
05ee41baf8
Add folder for mhlo::scatter
...
PiperOrigin-RevId: 337274351
2020-10-15 03:26:05 -07:00
0f36979e2c
Integrate LLVM at llvm/llvm-project@220de1f32a
...
Updates LLVM usage to match
[220de1f32add](https://github.com/llvm/llvm-project/commit/220de1f32add )
PiperOrigin-RevId: 337270545
2020-10-15 02:54:24 -07:00
e8eb3f80d1
Integrate LLVM at llvm/llvm-project@89657b3a3b
...
Updates LLVM usage to match
[89657b3a3b57](https://github.com/llvm/llvm-project/commit/89657b3a3b57 )
PiperOrigin-RevId: 337151511
2020-10-14 12:51:44 -07:00
f6b4e6758a
Add GPU specific LMHLO level ops
...
- Introduce operations in a new lmhlo_gpu dialect that map to GPU library function calls
in the XLA:GPU backend.
- Add basic unit tests as well.
PiperOrigin-RevId: 337132166
2020-10-14 11:23:55 -07:00
8506f1f26a
Integrate LLVM at llvm/llvm-project@9b3c2a72e4
...
Updates LLVM usage to match
[9b3c2a72e4cb](https://github.com/llvm/llvm-project/commit/9b3c2a72e4cb )
PiperOrigin-RevId: 337077072
2020-10-14 06:24:25 -07:00
8d69f52179
Integrate LLVM at llvm/llvm-project@d0c95808e5
...
Updates LLVM usage to match
[d0c95808e50c](https://github.com/llvm/llvm-project/commit/d0c95808e50c )
PiperOrigin-RevId: 337061504
2020-10-14 04:21:36 -07:00
3a4d99a3b4
Integrate LLVM at llvm/llvm-project@3b33b41604
...
Updates LLVM usage to match
[3b33b4160478](https://github.com/llvm/llvm-project/commit/3b33b4160478 )
PiperOrigin-RevId: 337052227
2020-10-14 02:58:14 -07:00
15fe5ca737
Integrate LLVM at llvm/llvm-project@6713332fdd
...
Updates LLVM usage to match
[6713332fddb7](https://github.com/llvm/llvm-project/commit/6713332fddb7 )
PiperOrigin-RevId: 336988895
2020-10-13 17:21:14 -07:00
53d16d9f8e
Integrate LLVM at llvm/llvm-project@93377888ae
...
Updates LLVM usage to match
[93377888ae89](https://github.com/llvm/llvm-project/commit/93377888ae89 )
PiperOrigin-RevId: 336669842
2020-10-12 08:42:31 -07:00
5039c00c4c
Integrate LLVM at llvm/llvm-project@473b364a19
...
Updates LLVM usage to match
[473b364a19ff](https://github.com/llvm/llvm-project/commit/473b364a19ff )
PiperOrigin-RevId: 336647877
2020-10-12 05:59:06 -07:00
2a1626bdfb
Integrate LLVM at llvm/llvm-project@a2291a58bf
...
Updates LLVM usage to match
[a2291a58bf1c](https://github.com/llvm/llvm-project/commit/a2291a58bf1c )
PiperOrigin-RevId: 336423546
2020-10-09 23:19:33 -07:00
ab4c03b84b
Replace operands now injected in OpBuilder
...
These are now injected automatically.
PiperOrigin-RevId: 336356284
2020-10-09 13:39:33 -07:00
f6af1fc134
Support hlo to lhlo buffer placement through shape.assuming ops.
...
PiperOrigin-RevId: 336287728
2020-10-09 07:14:12 -07:00
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
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
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
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
ac483f4091
Remove superfluous quote from lhlo_ops.td
...
PiperOrigin-RevId: 335921873
2020-10-07 12:06:36 -07:00
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
3736c5542f
[MLIR][KernelGen] Fix unranked codegeneration in kernel generator
...
PiperOrigin-RevId: 335847086
2020-10-07 05:39:55 -07:00
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
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
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
84277ed784
Remove unused `using` statements.
...
PiperOrigin-RevId: 335517711
2020-10-05 15:53:59 -07:00
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
a33441907c
[MLIR][KernelGen] Add E2E test for `tf.Acos`
...
PiperOrigin-RevId: 335419310
2020-10-05 08:14:36 -07:00
bae0815ef0
[MLIR][KernelGen] Legalize `atan` to approximation
...
PiperOrigin-RevId: 335417836
2020-10-05 08:05:52 -07:00
7f84a86cf5
[MLIR][KerneGen] Lower `tf.Atan` all the way to LLVM
...
PiperOrigin-RevId: 335394668
2020-10-05 05:07:13 -07:00
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
d927e32451
[HLO] Clean-up dynamic allocation in hlo-legalize-to-lhlo pass.
...
PiperOrigin-RevId: 335385243
2020-10-05 03:55:24 -07:00
7367eac074
Add folder for mhlo::remainder
...
PiperOrigin-RevId: 335372628
2020-10-05 02:20:01 -07:00
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
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
c708bfd6d0
[MLIR] Add cbrt, reduce-precision, and bitcast ops to MHLO.
...
PiperOrigin-RevId: 335109804
2020-10-02 15:13:18 -07:00
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
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
A. Unique TensorFlower
Jacques Pienaar
Stephan Herhut
Adrian Kuegel
Rahul Joshi
Tres Popp
Pavel Krajcevski
Lucy Fox
Alexander Belyaev
Tim Shen