Jacques Pienaar
2ea9470515
Remove BASE_HLO_ConvOp to remove coupling between MHLO and LMHLO conv ops
...
PiperOrigin-RevId: 373201247
2021-05-11 11:54:44 -07:00
Itai Zukerman
a4db6c57aa
Removed all (most) BASE_HLO_* ops.
...
Moved the corresponding `summary` and `description` fields into the subclasses.
Kept BASE_HLO_ConvOp for `hasWindowReversal()'.
PiperOrigin-RevId: 373173025
2021-05-11 09:48:31 -07:00
A. Unique TensorFlower
7f7a86ad0d
[MLIR][HLO] Implement `RegionBranchOpInterface` for rank specialization cluster
...
PiperOrigin-RevId: 373163196
2021-05-11 09:03:05 -07:00
Adrian Kuegel
1432db02e7
Integrate LLVM at llvm/llvm-project@df47368d40
...
Updates LLVM usage to match
[df47368d406a](https://github.com/llvm/llvm-project/commit/df47368d406a )
PiperOrigin-RevId: 373136440
2021-05-11 06:24:17 -07:00
A. Unique TensorFlower
96a47345cc
[MLIR][HLO] Add `rank_specialization_cluster` op to CHLO
...
The operation will be used to cluster compatible operations that can be rank-
specialized collectively.
PiperOrigin-RevId: 373128557
2021-05-11 05:17:42 -07:00
Benjamin Kramer
86b7eb434c
[MHLO] Don't crash trying to constant fold mhlo.convert on complex
...
MLIR still doesn't have a complex attribute so this can't be implemented, so
just bail out instead of trying to fold.
PiperOrigin-RevId: 373128307
2021-05-11 05:15:57 -07:00
Adrian Kuegel
3cce7017fc
Integrate LLVM at llvm/llvm-project@88a48999d2
...
Updates LLVM usage to match
[88a48999d249](https://github.com/llvm/llvm-project/commit/88a48999d249 )
PiperOrigin-RevId: 373119140
2021-05-11 03:58:30 -07:00
A. Unique TensorFlower
6191b3e528
[MLIR][HLO] Use `#` operator for list concatenation in C/HLO definitions
...
PiperOrigin-RevId: 373110780
2021-05-11 02:53:34 -07:00
Adrian Kuegel
920f261262
Integrate LLVM at llvm/llvm-project@888307ee62
...
Updates LLVM usage to match
[888307ee625b](https://github.com/llvm/llvm-project/commit/888307ee625b )
PiperOrigin-RevId: 373100219
2021-05-11 01:18:26 -07:00
A. Unique TensorFlower
1de55d2620
Integrate LLVM at llvm/llvm-project@1e11616a07
...
Updates LLVM usage to match
[1e11616a071d](https://github.com/llvm/llvm-project/commit/1e11616a071d )
PiperOrigin-RevId: 373087867
2021-05-10 23:34:27 -07:00
A. Unique TensorFlower
7f86dd9f7e
Constant fold compare EQ if one of the operands is true and compare NE if one of the operands is false.
...
PiperOrigin-RevId: 373058030
2021-05-10 18:53:49 -07:00
A. Unique TensorFlower
2d88f2f601
Integrate LLVM at llvm/llvm-project@bf812ea484
...
Updates LLVM usage to match
[bf812ea484b7](https://github.com/llvm/llvm-project/commit/bf812ea484b7 )
PiperOrigin-RevId: 373055329
2021-05-10 18:32:41 -07:00
A. Unique TensorFlower
2af1796194
Integrate LLVM at llvm/llvm-project@5c7b43aa82
...
Updates LLVM usage to match
[5c7b43aa8298](https://github.com/llvm/llvm-project/commit/5c7b43aa8298 )
PiperOrigin-RevId: 373028739
2021-05-10 15:46:34 -07:00
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