ccd70d5717
[MLIR][HLO] Add `rank-specialization-to-scf` pass
...
Currently the lowering is only implemented for the unary case. The n-ary case
will follow.
PiperOrigin-RevId: 374162772
2021-05-17 03:56:23 -07:00
295ef229d6
Integrate LLVM at llvm/llvm-project@a11117a7c0
...
Updates LLVM usage to match
[a11117a7c0a2](https://github.com/llvm/llvm-project/commit/a11117a7c0a2 )
PiperOrigin-RevId: 374152829
2021-05-17 02:33:42 -07:00
804d1b50d6
Integrate LLVM at llvm/llvm-project@88a8965a7d
...
Updates LLVM usage to match
[88a8965a7d93](https://github.com/llvm/llvm-project/commit/88a8965a7d93 )
PiperOrigin-RevId: 374099490
2021-05-16 17:25:22 -07:00
b06baae910
Fixing nondeterminism in pattern application.
...
The ReduceRegion* patterns are matching on the same ops as the PointwiseToLinalg*
patterns and on certain toolchains (MSVC) the order can be wrong. If the pointwise
runs first then it converts the op *within* the reduction before the reduction one
runs, leading to nested linalg op weirdness.
PiperOrigin-RevId: 373848269
2021-05-14 12:57:39 -07:00
a361253e4f
[HLO] Add custom print/parse for window attributes of convolutions (in LMHLO)
...
PiperOrigin-RevId: 373807616
2021-05-14 09:47:25 -07:00
e4caaaf921
Integrate LLVM at llvm/llvm-project@e51ef7f070
...
Updates LLVM usage to match
[e51ef7f0706a](https://github.com/llvm/llvm-project/commit/e51ef7f0706a )
PiperOrigin-RevId: 373772350
2021-05-14 06:12:47 -07:00
76341f3720
[MLIR][HLO] Add mixed test for `rank-specialization-cluster` pass
...
PiperOrigin-RevId: 373762814
2021-05-14 04:40:40 -07:00
9248f0a182
Integrate LLVM at llvm/llvm-project@c82a0ae70e
...
Updates LLVM usage to match
[c82a0ae70e28](https://github.com/llvm/llvm-project/commit/c82a0ae70e28 )
PiperOrigin-RevId: 373750063
2021-05-14 02:35:37 -07:00
b1b6e81913
Integrate LLVM at llvm/llvm-project@7c57a9bd7d
...
Updates LLVM usage to match
[7c57a9bd7d4c](https://github.com/llvm/llvm-project/commit/7c57a9bd7d4c )
PiperOrigin-RevId: 373739370
2021-05-14 00:56:40 -07:00
d2cc74317c
Implement constant folding for mhlo.Sign.
...
PiperOrigin-RevId: 373550014
2021-05-13 03:54:04 -07:00
d764806c1e
[MHLO:Linalg] Add support for lowering reshape of unsigned tensors
...
PiperOrigin-RevId: 373461627
2021-05-12 15:14:29 -07:00
a2c9b3c9d7
[mlir] update gentbl() Bazel macro
...
Rename `gentbl` to `gentbl_cc_library` to make it clearer which kind of rule is
ultimately used.
Update `gentbl_*` macros to take `tbl_outs` options as a list rather a
whitespace-separated string and remove the related string handling.
PiperOrigin-RevId: 373406352
2021-05-12 10:58:58 -07:00
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
A. Unique TensorFlower
Ben Vanik
Rahul Joshi
Adrian Kuegel
Hanhan Wang
Alex Zinenko
Itai Zukerman
Jacques Pienaar
Benjamin Kramer
Stephan Herhut