colin
/
mlir-hlo
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
634 Commits 1 Branch 0 Tags 2.6 MiB
Commit Graph

634 Commits

Author SHA1 Message Date
A. Unique TensorFlower 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
A. Unique TensorFlower 51cd4200b6 Make LMHLO's Dot have the same power as MHLO's DotGeneral. 
PiperOrigin-RevId: 337391565
 2020-10-15 15:09:06 -07:00
Adrian Kuegel 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
Adrian Kuegel 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
A. Unique TensorFlower 05ee41baf8 Add folder for mhlo::scatter 
PiperOrigin-RevId: 337274351
 2020-10-15 03:26:05 -07:00
Adrian Kuegel 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
A. Unique TensorFlower 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
Rahul Joshi 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
Adrian Kuegel 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
Adrian Kuegel 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
Adrian Kuegel 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
A. Unique TensorFlower 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
Adrian Kuegel 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
Adrian Kuegel 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
A. Unique TensorFlower 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
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