420c42a0a1
[MLIR][HLO] Support CHLO unary operations in rank specialization clustering
...
PiperOrigin-RevId: 373397321
2021-05-12 10:20:43 -07:00
596918a6f1
[MLIR][HLO] Allow rank specialization clustering with `chlo.broadcast_select` op
...
PiperOrigin-RevId: 373379990
2021-05-12 08:56:49 -07:00
e260aa771c
[HLO] Add custom print/parse for convolution dimension numbers (in LMHLO)
...
PiperOrigin-RevId: 373379227
2021-05-12 08:52:46 -07:00
30779f0c2f
Added build rules for HLO and LHLO ops MD.
...
PiperOrigin-RevId: 373373920
2021-05-12 08:23:40 -07:00
8895262330
Integrate LLVM at llvm/llvm-project@b7a11274f9
...
Updates LLVM usage to match
[b7a11274f90f](https://github.com/llvm/llvm-project/commit/b7a11274f90f )
PiperOrigin-RevId: 373353006
2021-05-12 06:04:40 -07:00
875803e5e1
[MLIR][HLO] Add more tests for `rank-specialization-cluster` pass
...
PiperOrigin-RevId: 373343750
2021-05-12 04:46:30 -07:00
b95162f182
Integrate LLVM at llvm/llvm-project@34c098b780
...
Updates LLVM usage to match
[34c098b780a2](https://github.com/llvm/llvm-project/commit/34c098b780a2 )
PiperOrigin-RevId: 373336846
2021-05-12 03:48:06 -07:00
313d24bc8f
[MLIR][HLO] Add `rank-specialization-cluster` pass
...
Add a pass to cluster unranked C/HLO operations in one
`chlo.rank_specialization_cluster` op. The C/HLO operations are moved to the
body of the operation. Later passes can use this to rank-specialize all these
operations together.
PiperOrigin-RevId: 373336725
2021-05-12 03:46:01 -07:00
7f84779868
Integrate LLVM at llvm/llvm-project@6351993da7
...
Updates LLVM usage to match
[6351993da72e](https://github.com/llvm/llvm-project/commit/6351993da72e )
PiperOrigin-RevId: 373327134
2021-05-12 02:34:59 -07:00
5ec03ba8a5
Integrate LLVM at llvm/llvm-project@23596fece0
...
Updates LLVM usage to match
[23596fece043](https://github.com/llvm/llvm-project/commit/23596fece043 )
PiperOrigin-RevId: 373315484
2021-05-12 00:56:04 -07:00
b783b3272e
Integrate LLVM at llvm/llvm-project@3f8be15f29
...
Updates LLVM usage to match
[3f8be15f2911](https://github.com/llvm/llvm-project/commit/3f8be15f2911 )
PiperOrigin-RevId: 373302768
2021-05-11 22:56:46 -07:00
3ccf3938fa
Integrate LLVM at llvm/llvm-project@fec2945998
...
Updates LLVM usage to match
[fec294599894](https://github.com/llvm/llvm-project/commit/fec294599894 )
PiperOrigin-RevId: 373275250
2021-05-11 18:37:01 -07:00
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
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
7f7a86ad0d
[MLIR][HLO] Implement `RegionBranchOpInterface` for rank specialization cluster
...
PiperOrigin-RevId: 373163196
2021-05-11 09:03:05 -07:00
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
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
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
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
6191b3e528
[MLIR][HLO] Use `#` operator for list concatenation in C/HLO definitions
...
PiperOrigin-RevId: 373110780
2021-05-11 02:53:34 -07:00
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
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
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
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
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
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
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
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
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
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
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
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
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
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
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
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
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
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
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
d8c40b691c
[MLIR][HLO] Add `shape.broadcast` canonicalization to unblock broadcast moving
...
PiperOrigin-RevId: 372120309
2021-05-05 07:16:49 -07:00
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
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
5a60793b31
[MHLO:Linalg] Add support for lowering dynamic-slice on unsigned ints
...
PiperOrigin-RevId: 371979004
2021-05-04 13:08:36 -07:00
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
bae9936b98
Replace !any_of with none_of
...
No functional change.
PiperOrigin-RevId: 371868221
2021-05-04 01:55:36 -07:00
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
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
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
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
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
A. Unique TensorFlower
Rahul Joshi
Itai Zukerman
Adrian Kuegel
Jacques Pienaar
Benjamin Kramer
Stephan Herhut
dfki-jugr
Geoffrey Martin-Noble
Tres Popp