Commit Graph

960 Commits

Author SHA1 Message Date
A. Unique TensorFlower 596918a6f1 [MLIR][HLO] Allow rank specialization clustering with `chlo.broadcast_select` op
PiperOrigin-RevId: 373379990
2021-05-12 08:56:49 -07:00
Rahul Joshi e260aa771c [HLO] Add custom print/parse for convolution dimension numbers (in LMHLO)
PiperOrigin-RevId: 373379227
2021-05-12 08:52:46 -07:00
Itai Zukerman 30779f0c2f Added build rules for HLO and LHLO ops MD.
PiperOrigin-RevId: 373373920
2021-05-12 08:23:40 -07:00
Adrian Kuegel 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
A. Unique TensorFlower 875803e5e1 [MLIR][HLO] Add more tests for `rank-specialization-cluster` pass
PiperOrigin-RevId: 373343750
2021-05-12 04:46:30 -07:00
Adrian Kuegel 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
A. Unique TensorFlower 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
Adrian Kuegel 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
A. Unique TensorFlower 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
A. Unique TensorFlower 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
A. Unique TensorFlower 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
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