Commit Graph

886 Commits

Author SHA1 Message Date
Adrian Kuegel 2a4c63d949 Integrate LLVM at llvm/llvm-project@9ba661f912
Updates LLVM usage to match
[9ba661f91276](https://github.com/llvm/llvm-project/commit/9ba661f91276)

PiperOrigin-RevId: 372895783
2021-05-10 03:57:06 -07:00
Adrian Kuegel 9d4c194c32 Integrate LLVM at llvm/llvm-project@5b158093e2
Updates LLVM usage to match
[5b158093e246](https://github.com/llvm/llvm-project/commit/5b158093e246)

PiperOrigin-RevId: 372868700
2021-05-10 00:16:48 -07:00
A. Unique TensorFlower 39195ba15c Integrate LLVM at llvm/llvm-project@808bc11d9e
Updates LLVM usage to match
[808bc11d9e1a](https://github.com/llvm/llvm-project/commit/808bc11d9e1a)

PiperOrigin-RevId: 372728035
2021-05-08 10:37:29 -07:00
Stephan Herhut 807d306229 Integrate LLVM at llvm/llvm-project@793b4b2603
Updates LLVM usage to match
[793b4b26039e](https://github.com/llvm/llvm-project/commit/793b4b26039e)

PiperOrigin-RevId: 372544965
2021-05-07 06:40:36 -07:00
Stephan Herhut 2fe6189703 Integrate LLVM at llvm/llvm-project@911a541620
Updates LLVM usage to match
[911a541620bc](https://github.com/llvm/llvm-project/commit/911a541620bc)

PiperOrigin-RevId: 372524822
2021-05-07 03:39:50 -07:00
Stephan Herhut 2d7278ee64 Integrate LLVM at llvm/llvm-project@44ee974e2f
Updates LLVM usage to match
[44ee974e2f3e](https://github.com/llvm/llvm-project/commit/44ee974e2f3e)

PiperOrigin-RevId: 372507420
2021-05-07 01:08:18 -07:00
Rahul Joshi ce4c76314e [NFC] Remove all_gather_dimension from all-to-all in the unit test
PiperOrigin-RevId: 372463706
2021-05-06 18:14:52 -07:00
A. Unique TensorFlower f8c629207c Integrate LLVM at llvm/llvm-project@a577d59db2
Updates LLVM usage to match
[a577d59db243](https://github.com/llvm/llvm-project/commit/a577d59db243)

PiperOrigin-RevId: 372449781
2021-05-06 16:40:30 -07:00
A. Unique TensorFlower 830adc235b Integrate LLVM at llvm/llvm-project@6660319cef
Updates LLVM usage to match
[6660319cef6e](https://github.com/llvm/llvm-project/commit/6660319cef6e)

PiperOrigin-RevId: 372388025
2021-05-06 11:36:30 -07:00
Rahul Joshi 8c854886cb [XLA:GPU] Allow all-gather operands to have different element types.
- XLA's all-gather combiner can create such all-gathers, so relax the same element type
  trait for all-gathers.

PiperOrigin-RevId: 372380446
2021-05-06 11:04:13 -07:00
Stephan Herhut 819adef41f Integrate LLVM at llvm/llvm-project@67cfefebbb
Updates LLVM usage to match
[67cfefebbbbb](https://github.com/llvm/llvm-project/commit/67cfefebbbbb)

PiperOrigin-RevId: 372362482
2021-05-06 09:42:50 -07:00
Adrian Kuegel b2bc17c8b0 Integrate LLVM at llvm/llvm-project@632ebc4ab4
Updates LLVM usage to match
[632ebc4ab437](https://github.com/llvm/llvm-project/commit/632ebc4ab437)

PiperOrigin-RevId: 372330771
2021-05-06 06:37:39 -07:00
A. Unique TensorFlower 42f87acd10 Integrate LLVM at llvm/llvm-project@0277a24f4b
Updates LLVM usage to match
[0277a24f4bba](https://github.com/llvm/llvm-project/commit/0277a24f4bba)

PiperOrigin-RevId: 372262890
2021-05-05 20:41:01 -07:00
Benjamin Kramer 1401bb3461 Integrate LLVM at llvm/llvm-project@bfb9c749c0
Updates LLVM usage to match
[bfb9c749c024](https://github.com/llvm/llvm-project/commit/bfb9c749c024)

PiperOrigin-RevId: 372163886
2021-05-05 11:05:46 -07:00
A. Unique TensorFlower d8c40b691c [MLIR][HLO] Add `shape.broadcast` canonicalization to unblock broadcast moving
PiperOrigin-RevId: 372120309
2021-05-05 07:16:49 -07:00
dfki-jugr 6bc854f5d9 PR #48667: [mlir-hlo] Added RegionBranchOpInterfaces to lmhlo operations.
Imported from GitHub PR https://github.com/tensorflow/tensorflow/pull/48667

Added RegionBranchOpInterfaces to lmhlo operations that use regions.
This is needed, since the bufferization features in MLIR have to reason about the control flow within these operations.
Copybara import of the project:

--
572fd7d850a46630b812da84e9094280f89f259e by Julian Gross <julian.gross@dfki.de>:

Added RegionBranchOpInterfaces to lmhlo operations.

PiperOrigin-RevId: 372070825
2021-05-05 00:27:56 -07:00
Geoffrey Martin-Noble ac68145565 [MHLO:Linalg] Add support for lowering concat of unsigned tensors
Nothing about concat here really. Just need to plumb through the type
conversion.

PiperOrigin-RevId: 372012957
2021-05-04 15:57:54 -07:00
Geoffrey Martin-Noble 5a60793b31 [MHLO:Linalg] Add support for lowering dynamic-slice on unsigned ints
PiperOrigin-RevId: 371979004
2021-05-04 13:08:36 -07:00
Stephan Herhut fdae82aac7 Integrate LLVM at llvm/llvm-project@34593ae998
Updates LLVM usage to match
[34593ae9982a](https://github.com/llvm/llvm-project/commit/34593ae9982a)

PiperOrigin-RevId: 371968431
2021-05-04 12:18:42 -07:00
Adrian Kuegel bae9936b98 Replace !any_of with none_of
No functional change.

PiperOrigin-RevId: 371868221
2021-05-04 01:55:36 -07:00
Adrian Kuegel 384b87fad0 Lower ReluGrad via chlo::BroadcastSelect.
This allows to get rid of the constraint that it needs to have a static shape.

PiperOrigin-RevId: 371862452
2021-05-04 01:03:02 -07:00
Stephan Herhut 4bf1904c86 Integrate LLVM at llvm/llvm-project@aad3113417
Updates LLVM usage to match
[aad311341736](https://github.com/llvm/llvm-project/commit/aad311341736)

PiperOrigin-RevId: 371716638
2021-05-03 09:49:06 -07:00
Stephan Herhut e6faaea280 Integrate LLVM at llvm/llvm-project@1f8963c801
Updates LLVM usage to match
[1f8963c80195](https://github.com/llvm/llvm-project/commit/1f8963c80195)

PiperOrigin-RevId: 371688312
2021-05-03 06:58:42 -07:00
A. Unique TensorFlower 8556fab4a9 Integrate LLVM at llvm/llvm-project@ad12590fdd
Updates LLVM usage to match
[ad12590fdd14](https://github.com/llvm/llvm-project/commit/ad12590fdd14)

PiperOrigin-RevId: 371430333
2021-04-30 17:04:09 -07:00
Tres Popp 3354640c3a Integrate LLVM at llvm/llvm-project@b2780cd744
Updates LLVM usage to match
[b2780cd744ea](https://github.com/llvm/llvm-project/commit/b2780cd744ea)

PiperOrigin-RevId: 371326420
2021-04-30 07:06:04 -07:00
Tres Popp c27590da7e Integrate LLVM at llvm/llvm-project@4750a8b1bc
Updates LLVM usage to match
[4750a8b1bcde](https://github.com/llvm/llvm-project/commit/4750a8b1bcde)

PiperOrigin-RevId: 371291231
2021-04-30 02:01:25 -07:00
A. Unique TensorFlower 4d7aa12caa Integrate LLVM at llvm/llvm-project@a76df78470
Updates LLVM usage to match
[a76df78470d7](https://github.com/llvm/llvm-project/commit/a76df78470d7)

PiperOrigin-RevId: 371244064
2021-04-29 18:09:39 -07:00
Benjamin Kramer f4414fcd66 [MHLO:Linalg] Add support for lowering unsigned ops
This strips away the signedness with a type converter, using unrealized
conversion casts. The rest is mostly mechanically pushing the original op down
the pipeline so lowerings can see the original types.

Signed types stay signless for now. This can be changed in the HLO bridge later.

I did a pass over all ops and added unsigned lowerings where they were missing.
There may be more.

Currently the lowering will die at a later stage because it doesn't understand
the unrealized casts.

PiperOrigin-RevId: 371077494
2021-04-29 02:27:35 -07:00
Tres Popp 09f6141562 Integrate LLVM at llvm/llvm-project@6d8d133862
Updates LLVM usage to match
[6d8d1338629c](https://github.com/llvm/llvm-project/commit/6d8d1338629c)

PiperOrigin-RevId: 371074366
2021-04-29 02:03:52 -07:00
A. Unique TensorFlower 4d41b11f3b Integrate LLVM at llvm/llvm-project@671f0e2e18
Updates LLVM usage to match
[671f0e2e189c](https://github.com/llvm/llvm-project/commit/671f0e2e189c)

PiperOrigin-RevId: 371011125
2021-04-28 16:37:53 -07:00
A. Unique TensorFlower 9e9434a0ea Integrate LLVM at llvm/llvm-project@bf9eef92b6
Updates LLVM usage to match
[bf9eef92b6cd](https://github.com/llvm/llvm-project/commit/bf9eef92b6cd)

PiperOrigin-RevId: 370802511
2021-04-27 18:06:33 -07:00
Prakalp Srivastava 565856acd1 Min and InfinityMin of boolean should be false.
PiperOrigin-RevId: 370759584
2021-04-27 14:11:18 -07:00
Benjamin Kramer b2a23bf269 Integrate LLVM at llvm/llvm-project@4b13b7581d
Updates LLVM usage to match
[4b13b7581db5](https://github.com/llvm/llvm-project/commit/4b13b7581db5)

PiperOrigin-RevId: 370736351
2021-04-27 12:19:05 -07:00
Tres Popp a5de7719d7 Integrate LLVM at llvm/llvm-project@f8de9aaef2
Updates LLVM usage to match
[f8de9aaef2f4](https://github.com/llvm/llvm-project/commit/f8de9aaef2f4)

PiperOrigin-RevId: 370631530
2021-04-27 01:53:38 -07:00
A. Unique TensorFlower 50287ac72c Integrate LLVM at llvm/llvm-project@494e77138c
Updates LLVM usage to match
[494e77138c2e](https://github.com/llvm/llvm-project/commit/494e77138c2e)

PiperOrigin-RevId: 370574378
2021-04-26 17:13:01 -07:00
A. Unique TensorFlower e500ab37a1 Introduce constant folds for ReduceOp with single LogicalAnd or LogicalOr op.
PiperOrigin-RevId: 370551483
2021-04-26 15:11:27 -07:00
A. Unique TensorFlower 1fff544339 Integrate LLVM at llvm/llvm-project@535df472b0
Updates LLVM usage to match
[535df472b042](https://github.com/llvm/llvm-project/commit/535df472b042)

PiperOrigin-RevId: 370502124
2021-04-26 11:26:16 -07:00
Adrian Kuegel 0e2b255f01 Lower LHLO::AbsOp to complex dialect.
Also fix the traits for LHLO::AbsOp to allow different types and add a
verifier.

PiperOrigin-RevId: 370438790
2021-04-26 05:44:03 -07:00
A. Unique TensorFlower 1c11075d62 Integrate LLVM at llvm/llvm-project@caea37b37e
Updates LLVM usage to match
[caea37b37e6a](https://github.com/llvm/llvm-project/commit/caea37b37e6a)

PiperOrigin-RevId: 370202306
2021-04-23 20:07:23 -07:00
A. Unique TensorFlower 0569b7f7a4 [MLIR][MHLO] Generalize extent tensor cast elimination in bcast moving
PiperOrigin-RevId: 370112887
2021-04-23 10:52:50 -07:00
A. Unique TensorFlower ab1ccbaa6e Integrate LLVM at llvm/llvm-project@91d3f73937
Updates LLVM usage to match
[91d3f73937b6](https://github.com/llvm/llvm-project/commit/91d3f73937b6)

PiperOrigin-RevId: 370089366
2021-04-23 08:56:00 -07:00
A. Unique TensorFlower 21e9365718 [MLIR][MHLO] Generalize extent tensor cast elimination in bcast moving
PiperOrigin-RevId: 370085141
2021-04-23 08:31:11 -07:00
A. Unique TensorFlower da5d252143 [MLIR] Merge extent tensor casts into `shape_of` ops in broadcast moving
PiperOrigin-RevId: 370058002
2021-04-23 04:44:01 -07:00
A. Unique TensorFlower 890a79641e Integrate LLVM at llvm/llvm-project@37e1458128
Updates LLVM usage to match
[37e145812855](https://github.com/llvm/llvm-project/commit/37e145812855)

PiperOrigin-RevId: 370020161
2021-04-22 22:57:08 -07:00
Hanhan Wang 49df46893c Add support for lowering variadic mhlo.reduce op.
Also add more lowering for body ops. Some MinOp and MaxOp can be legalized to
SelectOp + CompareOp.

PiperOrigin-RevId: 369891551
2021-04-22 09:50:49 -07:00
A. Unique TensorFlower 2dfea55d0c Integrate LLVM at llvm/llvm-project@2f69975683
Updates LLVM usage to match
[2f69975683f5](https://github.com/llvm/llvm-project/commit/2f69975683f5)

PiperOrigin-RevId: 369861751
2021-04-22 06:55:00 -07:00
A. Unique TensorFlower ca672cc0a4 Integrate LLVM at llvm/llvm-project@1a3f88658a
Updates LLVM usage to match
[1a3f88658a02](https://github.com/llvm/llvm-project/commit/1a3f88658a02)

PiperOrigin-RevId: 369789544
2021-04-21 20:22:09 -07:00
A. Unique TensorFlower a791d9cefc Integrate LLVM at llvm/llvm-project@4bb60c285c
Updates LLVM usage to match
[4bb60c285cb3](https://github.com/llvm/llvm-project/commit/4bb60c285cb3)

PiperOrigin-RevId: 369750761
2021-04-21 15:44:37 -07:00
A. Unique TensorFlower 57d7c94a24 Integrate LLVM at llvm/llvm-project@9430efa18b
Updates LLVM usage to match
[9430efa18b02](https://github.com/llvm/llvm-project/commit/9430efa18b02)

PiperOrigin-RevId: 369706935
2021-04-21 12:15:23 -07:00
Benjamin Kramer 4d435a817e [mhlo:linalg] Add support for lowering mhlo.concatenate to Linalg ops.
This uses a indexed linalg.generic, which is rather awkward standalone but
allows fusing into the output of the concatenate and avoid to ever materialize
it in memory. I think this is the only way to get that with the current linalg
stack, fusion across a concatenate would require more infrastructure.

PiperOrigin-RevId: 369677652
2021-04-21 10:01:08 -07:00