colin
/
mlir-hlo
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
981 Commits 1 Branch 0 Tags 2.6 MiB
Commit Graph

981 Commits

Author SHA1 Message Date
A. Unique TensorFlower 8db96f54d3 [mhlo] Add a folder for mhlo.map which does nothing but return one of the arguments. 
Add a folder for maps whose body returns only one of the arguments. When this arises the fold replaces the map output with one of the operand tensors.

PiperOrigin-RevId: 369304322
 2021-04-19 14:36:08 -07:00
A. Unique TensorFlower 9374a1c0c5 [MLIR] Fix merge of assuming ops 
Assuming ops can only be merged if their witnesses will dominate the merged
assuming op. This is not the case if the second op's witness is a result of the
first.

PiperOrigin-RevId: 369192868
 2021-04-19 04:21:08 -07:00
A. Unique TensorFlower 0bb866a799 Integrate LLVM at llvm/llvm-project@80e166f81a 
Updates LLVM usage to match
[80e166f81abd](https://github.com/llvm/llvm-project/commit/80e166f81abd)

PiperOrigin-RevId: 369176305
 2021-04-19 02:08:50 -07:00
A. Unique TensorFlower 411ee907e3 Integrate LLVM at llvm/llvm-project@517c3aee4d 
Updates LLVM usage to match
[517c3aee4de5](https://github.com/llvm/llvm-project/commit/517c3aee4de5)

PiperOrigin-RevId: 369017690
 2021-04-17 09:50:27 -07:00
A. Unique TensorFlower d938913704 Integrate LLVM at llvm/llvm-project@bb907b26e2 
Updates LLVM usage to match
[bb907b26e2bf](https://github.com/llvm/llvm-project/commit/bb907b26e2bf)

PiperOrigin-RevId: 368850216
 2021-04-16 08:21:21 -07:00
A. Unique TensorFlower 216447323a Integrate LLVM at llvm/llvm-project@3bc88eb392 
Updates LLVM usage to match
[3bc88eb3924f](https://github.com/llvm/llvm-project/commit/3bc88eb3924f)

PiperOrigin-RevId: 368828718
 2021-04-16 05:33:26 -07:00
A. Unique TensorFlower 4f8c0e6066 Integrate LLVM at llvm/llvm-project@3f97c66b00 
Updates LLVM usage to match
[3f97c66b0040](https://github.com/llvm/llvm-project/commit/3f97c66b0040)

PiperOrigin-RevId: 368763888
 2021-04-15 19:22:44 -07:00
A. Unique TensorFlower b1435bff87 Integrate LLVM at llvm/llvm-project@f28cb01be0 
Updates LLVM usage to match
[f28cb01be071](https://github.com/llvm/llvm-project/commit/f28cb01be071)

PiperOrigin-RevId: 368712932
 2021-04-15 14:02:42 -07:00
A. Unique TensorFlower a16b4f17e0 Integrate LLVM at llvm/llvm-project@93f54fae9d 
Updates LLVM usage to match
[93f54fae9dda](https://github.com/llvm/llvm-project/commit/93f54fae9dda)

PiperOrigin-RevId: 368656637
 2021-04-15 09:38:07 -07:00
A. Unique TensorFlower 0dcbe56bd7 Integrate LLVM at llvm/llvm-project@49cbf4cd85 
Updates LLVM usage to match
[49cbf4cd85a9](https://github.com/llvm/llvm-project/commit/49cbf4cd85a9)

PiperOrigin-RevId: 368630894
 2021-04-15 06:59:31 -07:00
A. Unique TensorFlower 2589ffe49f Integrate LLVM at llvm/llvm-project@71a45e7c63 
Updates LLVM usage to match
[71a45e7c6328](https://github.com/llvm/llvm-project/commit/71a45e7c6328)

PiperOrigin-RevId: 368611924
 2021-04-15 04:11:17 -07:00
A. Unique TensorFlower 3b3a61377e Integrate LLVM at llvm/llvm-project@c3f1271464 
Updates LLVM usage to match
[c3f127146415](https://github.com/llvm/llvm-project/commit/c3f127146415)

PiperOrigin-RevId: 368598603
 2021-04-15 02:24:43 -07:00
Adrian Kuegel db9f298505 Generate Equal and NotEqual kernels for complex types. 
PiperOrigin-RevId: 368586877
 2021-04-15 00:35:52 -07:00
A. Unique TensorFlower bfc8cca38f Integrate LLVM at llvm/llvm-project@8f9477b067 
Updates LLVM usage to match
[8f9477b067b0](https://github.com/llvm/llvm-project/commit/8f9477b067b0)

PiperOrigin-RevId: 368569183
 2021-04-14 21:50:44 -07:00
Rahul Joshi c75cbf4ac7 [MLIR][NFC] Rename ReduceOp operands() => inputs(). 
- Rename to avoid confusion as operands generally includes all operands of an operation

PiperOrigin-RevId: 368479524
 2021-04-14 12:08:23 -07:00
Prashant Kumar 236e7db5c0 PR #47315: [MLIR] Add concatenateOp lowering from lmhlo to Affine. 
Imported from GitHub PR https://github.com/tensorflow/tensorflow/pull/47315

Lowering of `concatenateOp` is added from lmhlo to Affine. The lowering
has been added as a part of `lhlo-legalize-to-affine` pass.

Signed-off-by: Prashant Kumar <prashantk@polymagelabs.com>
Copybara import of the project:

--
15314e4579f7a6901cf3475eff25962a34772eaf by Prashant Kumar <prashantk@polymagelabs.com>:

[MLIR] Add concatenateOp lowering from lmhlo to Affine.

Lowering of `concatenateOp` is added from lmhlo to Affine. The lowering
has been added as a part of `lhlo-legalize-to-affine` pass.

Signed-off-by: Prashant Kumar <prashantk@polymagelabs.com>
PiperOrigin-RevId: 368465992
 2021-04-14 11:06:38 -07:00
A. Unique TensorFlower c10167d4a8 Integrate LLVM at llvm/llvm-project@c4c9e4d6df 
Updates LLVM usage to match
[c4c9e4d6df3c](https://github.com/llvm/llvm-project/commit/c4c9e4d6df3c)

PiperOrigin-RevId: 368450496
 2021-04-14 09:54:28 -07:00
A. Unique TensorFlower 2db0e6280d Integrate LLVM at llvm/llvm-project@7975dd033c 
Updates LLVM usage to match
[7975dd033cb9](https://github.com/llvm/llvm-project/commit/7975dd033cb9)

PiperOrigin-RevId: 368399684
 2021-04-14 04:06:34 -07:00
A. Unique TensorFlower 79c7c48989 Integrate LLVM at llvm/llvm-project@32e264921b 
Updates LLVM usage to match
[32e264921b7a](https://github.com/llvm/llvm-project/commit/32e264921b7a)

PiperOrigin-RevId: 368352219
 2021-04-13 20:55:55 -07:00
Jacques Pienaar fdd75daed6 Add shape function for MHLO RngNormal and RngUniform 
PiperOrigin-RevId: 368276963
 2021-04-13 12:59:42 -07:00
A. Unique TensorFlower c01e96d095 Integrate LLVM at llvm/llvm-project@dad5caa59e 
Updates LLVM usage to match
[dad5caa59e6b](https://github.com/llvm/llvm-project/commit/dad5caa59e6b)

PiperOrigin-RevId: 368257770
 2021-04-13 11:28:38 -07:00
Hanhan Wang a3fc99efe0 Add support for lowering mhlo.dynamic_slice to Linalg ops. 
PiperOrigin-RevId: 368033540
 2021-04-12 10:34:55 -07:00
A. Unique TensorFlower 0ec0a23e61 [MLIR][HLO] Generalize merged witnesses in `move-up-dynamic-broadcasts-for-fusion` 
PiperOrigin-RevId: 368012460
 2021-04-12 08:55:29 -07:00
A. Unique TensorFlower 1007995ea2 Integrate LLVM at llvm/llvm-project@5a5a94ed34 
Updates LLVM usage to match
[5a5a94ed34b0](https://github.com/llvm/llvm-project/commit/5a5a94ed34b0)

PiperOrigin-RevId: 368012265
 2021-04-12 08:53:33 -07:00
A. Unique TensorFlower 7965f7fa79 Integrate LLVM at llvm/llvm-project@67b39661c8 
Updates LLVM usage to match
[67b39661c847](https://github.com/llvm/llvm-project/commit/67b39661c847)

PiperOrigin-RevId: 367984223
 2021-04-12 05:43:23 -07:00
A. Unique TensorFlower bc95a4652f Integrate LLVM at llvm/llvm-project@2dd22da965 
Updates LLVM usage to match
[2dd22da965ff](https://github.com/llvm/llvm-project/commit/2dd22da965ff)

PiperOrigin-RevId: 367965285
 2021-04-12 03:11:45 -07:00
Alexander Belyaev 8a9bf05d78 Integrate LLVM at llvm/llvm-project@6ce76ff7eb 
Updates LLVM usage to match
[6ce76ff7eb76](https://github.com/llvm/llvm-project/commit/6ce76ff7eb76)

PiperOrigin-RevId: 367678843
 2021-04-09 12:11:56 -07:00
A. Unique TensorFlower 6d2209e301 [MLIR][HLO] Canonicalize chained broadcasts 
Compose two subsequent `dynamic_broadcast_in_dim` ops into one.

PiperOrigin-RevId: 367630360
 2021-04-09 07:35:34 -07:00
Hanhan Wang fdb653788c Add support for lowering and/or within mhlo.reduce op body. 
PiperOrigin-RevId: 367627034
 2021-04-09 07:09:13 -07:00
Alexander Belyaev 2ef77c1c0c Integrate LLVM at llvm/llvm-project@39ae25fb8c 
Updates LLVM usage to match
[39ae25fb8c64](https://github.com/llvm/llvm-project/commit/39ae25fb8c64)

PiperOrigin-RevId: 367597627
 2021-04-09 02:35:49 -07:00
Hanhan Wang 768234b077 [NFC] Fix a typo in ScalarLimit comments. 
PiperOrigin-RevId: 367593932
 2021-04-09 01:57:50 -07:00
Alexander Belyaev bf487d51d4 Integrate LLVM at llvm/llvm-project@1b589172bd 
Updates LLVM usage to match
[1b589172bd19](https://github.com/llvm/llvm-project/commit/1b589172bd19)

PiperOrigin-RevId: 367482146
 2021-04-08 12:23:42 -07:00
Alexander Belyaev 39430e04da Integrate LLVM at llvm/llvm-project@b601c67192 
Updates LLVM usage to match
[b601c6719226](https://github.com/llvm/llvm-project/commit/b601c6719226)

PiperOrigin-RevId: 367413533
 2021-04-08 05:56:16 -07:00
Adrian Kuegel cc607bc72d Support up to rank 8 in rank specialization for SelectOp. 
PiperOrigin-RevId: 367406557
 2021-04-08 04:55:41 -07:00
Alexander Belyaev f068d26843 Integrate LLVM at llvm/llvm-project@6e8601ff4a 
Updates LLVM usage to match
[6e8601ff4ae1](https://github.com/llvm/llvm-project/commit/6e8601ff4ae1)

PiperOrigin-RevId: 367406032
 2021-04-08 04:52:04 -07:00
Alexander Belyaev 11c1bc4bd2 Integrate LLVM at llvm/llvm-project@628dda08b8 
Updates LLVM usage to match
[628dda08b82f](https://github.com/llvm/llvm-project/commit/628dda08b82f)

PiperOrigin-RevId: 367238750
 2021-04-07 09:47:33 -07:00
Alexander Belyaev f003df084b Integrate LLVM at llvm/llvm-project@c0ef93bec8 
Updates LLVM usage to match
[c0ef93bec85a](https://github.com/llvm/llvm-project/commit/c0ef93bec85a)

PiperOrigin-RevId: 367215596
 2021-04-07 07:30:33 -07:00
Hanhan Wang c466f08993 Add checks in ReduceWindowOpOnTensorsConversion. 
The pattern does not support ops with non-zero padding config. Add a check to
prevent unexpected lowering.

It is not easy to add tests because other patterns will convert body ops, and
it causes issues like invalid IRs.

PiperOrigin-RevId: 367202450
 2021-04-07 05:46:01 -07:00
Alexander Belyaev c6894333b9 Integrate LLVM at llvm/llvm-project@f37ea62e57 
Updates LLVM usage to match
[f37ea62e57b5](https://github.com/llvm/llvm-project/commit/f37ea62e57b5)

PiperOrigin-RevId: 367189460
 2021-04-07 03:56:23 -07:00
Alexander Belyaev f3295324e1 Integrate LLVM at llvm/llvm-project@000cf84cf1 
Updates LLVM usage to match
[000cf84cf1bb](https://github.com/llvm/llvm-project/commit/000cf84cf1bb)

PiperOrigin-RevId: 367179026
 2021-04-07 02:23:20 -07:00
A. Unique TensorFlower a62382cf1f Integrate LLVM at llvm/llvm-project@0e92cbd6a6 
Updates LLVM usage to match
[0e92cbd6a652](https://github.com/llvm/llvm-project/commit/0e92cbd6a652)

PiperOrigin-RevId: 367062980
 2021-04-06 12:40:51 -07:00
Rahul Joshi 0800423d27 [LMHLO] Simplify FusionOp::getInputBuffers() and friends. 
- No need to walk the entire region, instead just iterate over the top level operations in
  the region attached to the fusion op.

PiperOrigin-RevId: 366528833
 2021-04-02 15:55:49 -07:00
Geoffrey Martin-Noble 38d0f96709 Switch deps AllPassesAndDialectsNoRegistration -> AllPassesAndDialects 
These targets are now identical as all registration is explicit.
Temporarily leaving the old target as a (deprecated) alias while
changes propagate.

PiperOrigin-RevId: 366513211
 2021-04-02 14:26:13 -07:00
A. Unique TensorFlower c315b07d9c Integrate LLVM at llvm/llvm-project@8e5f3d04f2 
Updates LLVM usage to match
[8e5f3d04f269](https://github.com/llvm/llvm-project/commit/8e5f3d04f269)

PiperOrigin-RevId: 366508804
 2021-04-02 14:02:15 -07:00
A. Unique TensorFlower ad442fd99a Integrate LLVM at llvm/llvm-project@17800f900d 
Updates LLVM usage to match
[17800f900dca](https://github.com/llvm/llvm-project/commit/17800f900dca)

PiperOrigin-RevId: 366356934
 2021-04-01 16:46:09 -07:00
Geoffrey Martin-Noble 763ff55970 Restore SingleBlockImplicitTerminator verification to mhlo.while 
The internal users have been cleaned up, so we can roll this forward again.

PiperOrigin-RevId: 366313960
 2021-04-01 13:04:26 -07:00
A. Unique TensorFlower f1bed7b340 Integrate LLVM at llvm/llvm-project@d61b40ed27 
Updates LLVM usage to match
[d61b40ed2750](https://github.com/llvm/llvm-project/commit/d61b40ed2750)

PiperOrigin-RevId: 366288494
 2021-04-01 11:05:39 -07:00
Rahul Joshi ff2cbfa2ec [MLIR] Add support for representing variadic reduce-window in HLO/LMHLO dialect. 
-  Fixed a subset of transformations to handle variadic reduce-window.

PiperOrigin-RevId: 366278650
 2021-04-01 10:24:50 -07:00
A. Unique TensorFlower d1f697e618 Integrate LLVM at llvm/llvm-project@abbe80fa52 
Updates LLVM usage to match
[abbe80fa52c5](https://github.com/llvm/llvm-project/commit/abbe80fa52c5)

PiperOrigin-RevId: 366236030
 2021-04-01 05:58:20 -07:00
A. Unique TensorFlower c23be1841c [MLIR] Add example test case for `move-up-dynamic-broadcasts-for-fusion` pass 
Add exemplary test case as it appears in the lowering of two subsequent `tf.Sub`
ops.

PiperOrigin-RevId: 366219139
 2021-04-01 03:24:43 -07:00