a7a7184eb6
[XLA][MLIR] Lower `tf.Tan` and `tf.Sin` to MLHLO
...
Add `tan` op and lowering to CHLO dialect, move CHLO lowerings to
`chlo_legalize_to_hlo_patterns` and extend missing patterns.
PiperOrigin-RevId: 331128170
2020-09-11 05:05:58 -07:00
90927f6b53
[XLA][MLIR] Lower `tf.Tan` and `tf.Sin` to MLHLO
...
Add `tan` op and lowering to CHLO dialect, move CHLO lowerings to
`chlo_legalize_to_hlo_patterns` and extend missing patterns.
PiperOrigin-RevId: 331125286
2020-09-11 04:39:28 -07:00
49aaaed762
Integrate LLVM at llvm/llvm-project@be7cef789e
...
Updates LLVM usage to match
[be7cef789e75](https://github.com/llvm/llvm-project/commit/be7cef789e75 )
PiperOrigin-RevId: 331065004
2020-09-10 19:11:17 -07:00
d0c8d17373
Update mhlo.concatenate inferred return type for dynamics
...
MHLO concatenate should support dynamic inputs. Its possible that the output
shape can be inferred from a dimension in one input that is not dynamic in
another.
PiperOrigin-RevId: 331054181
2020-09-10 17:46:11 -07:00
6eefb07767
Fix zero attr mistake
...
PiperOrigin-RevId: 331050555
2020-09-10 17:21:39 -07:00
b22f2f0eea
Integrate LLVM at llvm/llvm-project@52f0837778
...
Updates LLVM usage to match
[52f0837778b6](https://github.com/llvm/llvm-project/commit/52f0837778b6 )
PiperOrigin-RevId: 330939173
2020-09-10 08:14:23 -07:00
d599485e06
PR #43069 : [hlo] Unbreak hlo-legalize-to-lhlo test
...
Imported from GitHub PR https://github.com/tensorflow/tensorflow/pull/43069
The legalization of mlhlo.ReturnOp to lhlo.TerminatorOp by using BufferAssignmentReturnOpConverter fails since the Memref typed results (or the Memref typed operands of Return operation) are set to stay as results after legalization but lhlo.TerminatorOp doesn't accept any operands. Therefore, BufferAssignmentReturnOpConverter must be replaced with a manual conversion that removes all operands of mlhlo.ReturnOp and inserts copy operations in their places.
Copybara import of the project:
--
8be0435b0147263c3872bedec58fd215f784b450 by Ehsan Toosi <ehsan.nadjaran_toosi@dfki.de>:
[hlo] Unbreak hlo-legalize-to-lhlo test
COPYBARA_INTEGRATE_REVIEW=https://github.com/tensorflow/tensorflow/pull/43069 from dfki-ehna:fix_hlo_legalize_to_lhlo_test 8be0435b0147263c3872bedec58fd215f784b450
PiperOrigin-RevId: 330907602
2020-09-10 04:09:03 -07:00
f46ba09653
Integrate LLVM at llvm/llvm-project@4964d75d70
...
Updates LLVM usage to match
[4964d75d7078](https://github.com/llvm/llvm-project/commit/4964d75d7078 )
PiperOrigin-RevId: 330713009
2020-09-09 06:50:26 -07:00
81d51d810b
Bitwise and/or/xor folders
...
Includes both and/or/xor on same inputs, constant all ones/zeros single arg
folder, and constant input folders.
PiperOrigin-RevId: 330610858
2020-09-08 16:27:13 -07:00
9d4273b5a7
Delete mhlo static dialect registration
...
This is obsolete now.
PiperOrigin-RevId: 330605210
2020-09-08 15:57:36 -07:00
062a3ac4a0
Integrate LLVM at llvm/llvm-project@8d9c13f37d
...
Updates LLVM usage to match
[8d9c13f37d20](https://github.com/llvm/llvm-project/commit/8d9c13f37d20 )
PiperOrigin-RevId: 330496008
2020-09-08 06:20:11 -07:00
ebc7992d31
[MLIR][KERNEL_GEN] Add a library to lower kernels with the host side.
...
* Unified TF->Cubin and TF->Kernel_with_host side lowering in `kernel_creator.h|cc`
* Added a pass that attaches GPU binary blob to GPUModuleOp
* Refactored most of the code.
* Added tf_to_kernel binary that emits obj file
PiperOrigin-RevId: 330494488
2020-09-08 06:06:29 -07:00
64d9a465ee
Integrate LLVM at llvm/llvm-project@a70f2eb3e3
...
Updates LLVM usage to match
[a70f2eb3e39a](https://github.com/llvm/llvm-project/commit/a70f2eb3e39a )
PiperOrigin-RevId: 330468316
2020-09-08 02:03:22 -07:00
e391762513
Integrate LLVM at llvm/llvm-project@7cfc8f0c7c
...
Updates LLVM usage to match
[7cfc8f0c7c24](https://github.com/llvm/llvm-project/commit/7cfc8f0c7c24 )
PiperOrigin-RevId: 330443088
2020-09-07 21:14:16 -07:00
8e2b874fe8
Integrate LLVM at llvm/llvm-project@7d53fecb67
...
Updates LLVM usage to match
[7d53fecb6792](https://github.com/llvm/llvm-project/commit/7d53fecb6792 )
PiperOrigin-RevId: 330418199
2020-09-07 15:14:42 -07:00
73b4861f2c
Canonicalize mhlo.gather to mhlo.slice if it has a single set of constant indices
...
PiperOrigin-RevId: 330380755
2020-09-07 07:28:46 -07:00
dde1ed56cc
Add support for legalizing mhlo.slice to lmhlo.slice
...
PiperOrigin-RevId: 330153599
2020-09-04 21:43:01 -07:00
2a9d7ac084
Add support for legalizing lmhlo.transpose to linalg.generic
...
PiperOrigin-RevId: 330130704
2020-09-04 15:37:42 -07:00
f5d12604ed
Add support for legalizing mhlo.transpose to lmhlo.transpose
...
PiperOrigin-RevId: 330127758
2020-09-04 14:59:08 -07:00
b6b30698ad
Integrate LLVM at llvm/llvm-project@2dd9a4d855
...
Updates LLVM usage to match
[2dd9a4d855f6](https://github.com/llvm/llvm-project/commit/2dd9a4d855f6 )
PiperOrigin-RevId: 330048272
2020-09-03 21:28:38 -07:00
b7248424ae
Fix return type of replica-id to unsigned int 32 tensor.
...
PiperOrigin-RevId: 330012181
2020-09-03 16:08:38 -07:00
12b221f459
Integrate LLVM at llvm/llvm-project@dc8d7d23d8
...
Updates LLVM usage to match
[dc8d7d23d8d2](https://github.com/llvm/llvm-project/commit/dc8d7d23d8d2 )
PiperOrigin-RevId: 329992306
2020-09-03 14:21:30 -07:00
56689ce908
Temporarily mark hlo-legalize-to-lhlo.mlir as failing, pending fixes
...
Upstream changed the buffer assignment API and this needs adjustment on our side.
PiperOrigin-RevId: 329959756
2020-09-03 11:36:20 -07:00
801262688c
Integrate LLVM at llvm/llvm-project@1426ac0482
...
Updates LLVM usage to match
[1426ac048295](https://github.com/llvm/llvm-project/commit/1426ac048295 )
PiperOrigin-RevId: 329839640
2020-09-02 20:03:41 -07:00
aedcea6acf
Integrate LLVM at llvm/llvm-project@3445ec9ba7
...
Updates LLVM usage to match
[3445ec9ba718](https://github.com/llvm/llvm-project/commit/3445ec9ba718 )
PiperOrigin-RevId: 329691486
2020-09-02 05:08:34 -07:00
65b0613491
Integrate LLVM at llvm/llvm-project@202766947e
...
Updates LLVM usage to match
[202766947edb](https://github.com/llvm/llvm-project/commit/202766947edb )
PiperOrigin-RevId: 329673065
2020-09-02 02:27:52 -07:00
7c93352a40
Scalar / Trivial folding for mhlo.select
...
This covers the case where the predicate is a splat or the on_true/on_false
values are the same.
PiperOrigin-RevId: 329622785
2020-09-01 18:34:12 -07:00
158dbba4e5
Integrate LLVM at llvm/llvm-project@5ffd940ac0
...
Updates LLVM usage to match
[5ffd940ac02a](https://github.com/llvm/llvm-project/commit/5ffd940ac02a )
PiperOrigin-RevId: 329606435
2020-09-01 16:40:41 -07:00
a622bf479b
Fix mhlo::SliceOp::fold to not crash on unknown shapes
...
PiperOrigin-RevId: 329504383
2020-09-01 07:37:36 -07:00
dcab119cec
Integrate LLVM at llvm/llvm-project@ffd0b31c7c
...
Updates LLVM usage to match
[ffd0b31c7cba](https://github.com/llvm/llvm-project/commit/ffd0b31c7cba )
PiperOrigin-RevId: 329466048
2020-09-01 02:06:53 -07:00
0e6e2f06e7
Integrate LLVM at llvm/llvm-project@1d3d9b9cd8
...
Updates LLVM usage to match
[1d3d9b9cd808](https://github.com/llvm/llvm-project/commit/1d3d9b9cd808 )
PiperOrigin-RevId: 329432069
2020-08-31 20:34:04 -07:00
3e870929b5
Integrate LLVM at llvm/llvm-project@646f19bb9d
...
Updates LLVM usage to match
[646f19bb9dc8](https://github.com/llvm/llvm-project/commit/646f19bb9dc8 )
PiperOrigin-RevId: 329406157
2020-08-31 16:55:04 -07:00
a4857fad84
Integrate LLVM at llvm/llvm-project@bc3d4d9ed7
...
Updates LLVM usage to match
[bc3d4d9ed783](https://github.com/llvm/llvm-project/commit/bc3d4d9ed783 )
PiperOrigin-RevId: 329359501
2020-08-31 12:54:51 -07:00
049f034116
Lower mhlo.floor to lmhlo to linalg
...
PiperOrigin-RevId: 329304882
2020-08-31 08:16:36 -07:00
acf7413d40
Integrate LLVM at llvm/llvm-project@1d01fc100b
...
Updates LLVM usage to match
[1d01fc100bb5](https://github.com/llvm/llvm-project/commit/1d01fc100bb5 )
PiperOrigin-RevId: 329266181
2020-08-31 02:47:45 -07:00
f619da8a5c
Integrate LLVM at llvm/llvm-project@11cf6346fd
...
Updates LLVM usage to match
[11cf6346fd49](https://github.com/llvm/llvm-project/commit/11cf6346fd49 )
PiperOrigin-RevId: 329251819
2020-08-31 00:26:30 -07:00
29d7f01fbb
Integrate LLVM at llvm/llvm-project@90166c2563
...
Updates LLVM usage to match
[90166c256310](https://github.com/llvm/llvm-project/commit/90166c256310 )
PiperOrigin-RevId: 329065383
2020-08-28 23:07:36 -07:00
344c500fca
[mhlo] Add legalize to SCF pass
...
Start of pass to legalize MHLO control flow to SCF for further optimization in common form. The current version just matches a very simple instance (which also happens to occur a few times). Exposes some further canonicalization opportunities that aren't yet addressed.
PiperOrigin-RevId: 329017723
2020-08-28 15:11:58 -07:00
7176fb1839
Integrate LLVM at llvm/llvm-project@85dacca29f
...
Updates LLVM usage to match
[85dacca29f82](https://github.com/llvm/llvm-project/commit/85dacca29f82 )
PiperOrigin-RevId: 328919376
2020-08-28 04:58:12 -07:00
848ca244d2
Integrate LLVM at llvm/llvm-project@b5924a8e27
...
Updates LLVM usage to match
[b5924a8e2753](https://github.com/llvm/llvm-project/commit/b5924a8e2753 )
PiperOrigin-RevId: 328760781
2020-08-27 10:12:46 -07:00
011ecf41d3
Integrate LLVM at llvm/llvm-project@1d8af682ef
...
Updates LLVM usage to match
[1d8af682ef1d](https://github.com/llvm/llvm-project/commit/1d8af682ef1d )
PiperOrigin-RevId: 328733270
2020-08-27 07:28:58 -07:00
ca33164bb6
Integrate LLVM at llvm/llvm-project@95848ea101
...
Updates LLVM usage to match
[95848ea10127](https://github.com/llvm/llvm-project/commit/95848ea10127 )
PiperOrigin-RevId: 328605350
2020-08-26 14:16:41 -07:00
e8db30a663
Integrate LLVM at llvm/llvm-project@642cb7865f
...
Updates LLVM usage to match
[642cb7865f35](https://github.com/llvm/llvm-project/commit/642cb7865f35 )
PiperOrigin-RevId: 328515095
2020-08-26 05:48:57 -07:00
36ddbeb6b2
Remove the dependency on global dialect registry from mlir-hlo
...
PiperOrigin-RevId: 328457105
2020-08-25 20:30:42 -07:00
ebe7857fb1
Integrate LLVM at llvm/llvm-project@9500a72091
...
Updates LLVM usage to match
[9500a7209163](https://github.com/llvm/llvm-project/commit/9500a7209163 )
PiperOrigin-RevId: 328436293
2020-08-25 17:24:17 -07:00
7bc22bfa11
Integrate LLVM at llvm/llvm-project@ef76686916
...
Updates LLVM usage to match
[ef76686916d4](https://github.com/llvm/llvm-project/commit/ef76686916d4 )
PiperOrigin-RevId: 328404846
2020-08-25 14:33:30 -07:00
dd5f6fb550
Integrate LLVM at llvm/llvm-project@f8454d60b8
...
Updates LLVM usage to match
[f8454d60b829](https://github.com/llvm/llvm-project/commit/f8454d60b829 )
PiperOrigin-RevId: 328352040
2020-08-25 10:07:12 -07:00
3ce5337881
Integrate LLVM at llvm/llvm-project@c6fb72de4f
...
Updates LLVM usage to match
[c6fb72de4f55](https://github.com/llvm/llvm-project/commit/c6fb72de4f55 )
PiperOrigin-RevId: 328304428
2020-08-25 04:05:03 -07:00
e4383a1da8
Integrate LLVM at llvm/llvm-project@77e5a195f8
...
Updates LLVM usage to match
[77e5a195f818](https://github.com/llvm/llvm-project/commit/77e5a195f818 )
PiperOrigin-RevId: 328236932
2020-08-24 17:12:41 -07:00
94296bb7ec
PR #42509 : [MLIR] Add folder for mhlo get_dimension_size
...
Imported from GitHub PR https://github.com/tensorflow/tensorflow/pull/42509
Add folder for mhlo GetDimensionSizeOp (`mhlo.get_dimension_size`).
`get_dimension_size` folds to a constant when the corresponding tensor
dimension size is statically known / constant.
Copybara import of the project:
--
5994915525ec2e932125aa1f133ce2260ba100af by Uday Bondhugula <uday@polymagelabs.com>:
[MLIR] Add folder for mhlo get_dimension_size
Add folder for mhlo GetDimensionSizeOp. get_dimension_size folds to a
constant when the corresponding tensor dimension size is statically
known / constant.
COPYBARA_INTEGRATE_REVIEW=https://github.com/tensorflow/tensorflow/pull/42509 from polymage-labs:get_dimension_size_fold 5994915525ec2e932125aa1f133ce2260ba100af
PiperOrigin-RevId: 328222517
2020-08-24 15:36:30 -07:00
A. Unique TensorFlower
Robert Suderman
Ehsan Toosi
Mehdi Amini
Alexander Belyaev
Eugene Burmako
Prakalp Srivastava
Benjamin Kramer
Jacques Pienaar
Uday Bondhugula