4a29ca3b1d
Add layout to mhlo::InfeedOp td.
...
PiperOrigin-RevId: 356286875
2021-02-08 09:48:14 -08:00
80d753c1fe
Integrate LLVM at llvm/llvm-project@f89f6d1e5d
...
Updates LLVM usage to match
[f89f6d1e5d7d](https://github.com/llvm/llvm-project/commit/f89f6d1e5d7d )
PiperOrigin-RevId: 356265374
2021-02-08 09:47:00 -08:00
d086b8a0ec
Correct HLO atan2 lowering in cases of -inf and -0 inputs.
...
This is being done by just removing the approximation and lowering to atan2 lib calls later to make the implementation the same as XLA. Note that if the approximation is brought back later, it can be fixed by changing the IR checking `less-than(X, 0)` to `less-than(copysign(X, 1), 0)`
PiperOrigin-RevId: 356253941
2021-02-08 06:58:04 -08:00
bd0856578f
Integrate LLVM at llvm/llvm-project@a4fa667dee
...
Updates LLVM usage to match
[a4fa667dee60](https://github.com/llvm/llvm-project/commit/a4fa667dee60 )
PiperOrigin-RevId: 355945749
2021-02-05 15:53:55 -08:00
2aa8a90c69
Integrate LLVM at llvm/llvm-project@a1a1d338e9
...
Updates LLVM usage to match
[a1a1d338e99d](https://github.com/llvm/llvm-project/commit/a1a1d338e99d )
PiperOrigin-RevId: 355927079
2021-02-05 14:20:29 -08:00
b251712b1d
[XLA:GPU] Add conversion from HLO -> MLIR LMHLO for TriangularSolve
...
- Also add layout attributes for inputs and output for error checking.
PiperOrigin-RevId: 355863625
2021-02-05 09:18:02 -08:00
1c4521cc42
Integrate LLVM at llvm/llvm-project@d1978fa4bf
...
Updates LLVM usage to match
[d1978fa4bf0d](https://github.com/llvm/llvm-project/commit/d1978fa4bf0d )
PiperOrigin-RevId: 355848094
2021-02-05 07:42:06 -08:00
99bc05f2e4
Integrate LLVM at llvm/llvm-project@91e7a17133
...
Updates LLVM usage to match
[91e7a1713332](https://github.com/llvm/llvm-project/commit/91e7a1713332 )
PiperOrigin-RevId: 355702100
2021-02-04 13:42:31 -08:00
60e1b6882c
Add kernel definition for zeta operation.
...
PiperOrigin-RevId: 355575619
2021-02-04 01:27:43 -08:00
945bf8768d
Integrate LLVM at llvm/llvm-project@56fcd4ea8d
...
Updates LLVM usage to match
[56fcd4ea8daf](https://github.com/llvm/llvm-project/commit/56fcd4ea8daf )
PiperOrigin-RevId: 355529409
2021-02-03 18:48:22 -08:00
3380d0a761
Integrate LLVM at llvm/llvm-project@ea5b75de49
...
Updates LLVM usage to match
[ea5b75de4999](https://github.com/llvm/llvm-project/commit/ea5b75de4999 )
PiperOrigin-RevId: 355493884
2021-02-03 15:20:56 -08:00
44d0464d16
Use linalg.fill on tensors instead of tensor.generate in MHLO -> Linalg conversion.
...
linalg.fill on tensors is a structured op that allows use tile + fuse
to reduce the fill overhead.
PiperOrigin-RevId: 355490400
2021-02-03 15:03:49 -08:00
10cd797d6d
Integrate LLVM at llvm/llvm-project@bbb7555403
...
Updates LLVM usage to match
[bbb755540385](https://github.com/llvm/llvm-project/commit/bbb755540385 )
PiperOrigin-RevId: 355462338
2021-02-03 12:58:16 -08:00
6cd1875ee4
Implement lowering of chlo::zeta to mhlo dialect.
...
PiperOrigin-RevId: 355395581
2021-02-03 07:50:05 -08:00
9d8a53c452
Integrate LLVM at llvm/llvm-project@e635feb15a
...
Updates LLVM usage to match
[e635feb15a91](https://github.com/llvm/llvm-project/commit/e635feb15a91 )
PiperOrigin-RevId: 355392465
2021-02-03 07:32:31 -08:00
0472837e50
Integrate LLVM at llvm/llvm-project@54afcade3b
...
Updates LLVM usage to match
[54afcade3bbc](https://github.com/llvm/llvm-project/commit/54afcade3bbc )
PiperOrigin-RevId: 355378020
2021-02-03 05:50:41 -08:00
1b63101a38
Integrate LLVM at llvm/llvm-project@fae6d129da
...
Updates LLVM usage to match
[fae6d129dac2](https://github.com/llvm/llvm-project/commit/fae6d129dac2 )
PiperOrigin-RevId: 355360042
2021-02-03 03:21:07 -08:00
39aa1235d1
Integrate LLVM at llvm/llvm-project@2668714747
...
Updates LLVM usage to match
[2668714747c5](https://github.com/llvm/llvm-project/commit/2668714747c5 )
PiperOrigin-RevId: 355292910
2021-02-02 17:58:41 -08:00
7cbf8d8878
Integrate LLVM at llvm/llvm-project@83d705adb2
...
Updates LLVM usage to match
[83d705adb2e0](https://github.com/llvm/llvm-project/commit/83d705adb2e0 )
PiperOrigin-RevId: 355258427
2021-02-02 14:53:50 -08:00
04110a4b1c
Integrate LLVM at llvm/llvm-project@67dfe9c8d7
...
Updates LLVM usage to match
[67dfe9c8d70c](https://github.com/llvm/llvm-project/commit/67dfe9c8d70c )
PiperOrigin-RevId: 355235205
2021-02-02 13:09:20 -08:00
3b67b207c4
[MLIR][CHLO] Use CHLO lowering for `is_inf` op
...
PiperOrigin-RevId: 355189054
2021-02-02 09:53:13 -08:00
0726ab20fa
Integrate LLVM at llvm/llvm-project@4d904776a7
...
Updates LLVM usage to match
[4d904776a77a](https://github.com/llvm/llvm-project/commit/4d904776a77a )
PiperOrigin-RevId: 355142758
2021-02-02 05:11:40 -08:00
ae722a883f
Improve performance of lowered chlo.pow with integers
...
The new lowering takes 6 iterations of a loop always rather than iterating the exponent's number of times.
PiperOrigin-RevId: 355131133
2021-02-02 03:28:38 -08:00
0458ae9a22
[MLIR][KernelGen] Add `tf.Digamma` kernels
...
PiperOrigin-RevId: 355129028
2021-02-02 03:07:39 -08:00
f40ccc5b4b
[MLIR][CHLO] Add `chlo.digamma` and lowering to MHLO
...
PiperOrigin-RevId: 355122765
2021-02-02 02:10:17 -08:00
c2115f56c7
Integrate LLVM at llvm/llvm-project@8f7f2c4211
...
Updates LLVM usage to match
[8f7f2c4211ca](https://github.com/llvm/llvm-project/commit/8f7f2c4211ca )
PiperOrigin-RevId: 355120697
2021-02-02 01:54:32 -08:00
96f8771ed7
Add MLIR generated kernel for Angle kernel.
...
This also requires a canonicalization pattern to remove a redundant dynamic
reshape from rank 1 to rank 1.
PiperOrigin-RevId: 355113135
2021-02-02 00:47:20 -08:00
9d682343a9
Integrate LLVM at llvm/llvm-project@68f66f37d7
...
Updates LLVM usage to match
[68f66f37d7d7](https://github.com/llvm/llvm-project/commit/68f66f37d7d7 )
PiperOrigin-RevId: 355069626
2021-02-01 18:17:47 -08:00
6c8ad0895b
Integrate LLVM at llvm/llvm-project@75347ba1fa
...
Updates LLVM usage to match
[75347ba1fae2](https://github.com/llvm/llvm-project/commit/75347ba1fae2 )
PiperOrigin-RevId: 355031910
2021-02-01 14:51:46 -08:00
16bb35009b
Integrate LLVM at llvm/llvm-project@78c22fbce9
...
Updates LLVM usage to match
[78c22fbce991](https://github.com/llvm/llvm-project/commit/78c22fbce991 )
PiperOrigin-RevId: 354999980
2021-02-01 12:28:24 -08:00
816d279be3
[MLIR][CHLO] Simplify conversions with upcast
...
PiperOrigin-RevId: 354975366
2021-02-01 10:48:10 -08:00
8e3890e8e8
[MLIR:HLO] Add AllGather and AllToAll operations to LMHLO dialect.
...
- Use a common base class to for AllReduce, AllGather, and AllToAll in the ODS spec.
- Add basic verification for replica groups attribute.
PiperOrigin-RevId: 354969654
2021-02-01 10:23:46 -08:00
7a251aaed4
Integrate LLVM at llvm/llvm-project@49a6502cd5
...
Updates LLVM usage to match
[49a6502cd5c2](https://github.com/llvm/llvm-project/commit/49a6502cd5c2 )
PiperOrigin-RevId: 354936788
2021-02-01 07:43:37 -08:00
2da9d6947d
Integrate LLVM at llvm/llvm-project@e90e455d2a
...
Updates LLVM usage to match
[e90e455d2a0c](https://github.com/llvm/llvm-project/commit/e90e455d2a0c )
PiperOrigin-RevId: 354922438
2021-02-01 05:57:58 -08:00
1be1123c70
[XLA:GPU] Add support for PartitionId
...
PiperOrigin-RevId: 354599221
2021-01-29 13:31:54 -08:00
b1ce05cfc9
Integrate LLVM at llvm/llvm-project@1c762a81d2
...
Updates LLVM usage to match
[1c762a81d20f](https://github.com/llvm/llvm-project/commit/1c762a81d20f )
PiperOrigin-RevId: 354567452
2021-01-29 11:04:26 -08:00
2b72ddc6b2
[MLIR][KernelGen] Add `lgamma` kernels
...
PiperOrigin-RevId: 354519407
2021-01-29 06:14:17 -08:00
e61ef86fdb
Add zeta and broadcasting_zeta to chlo dialect.
...
PiperOrigin-RevId: 354500879
2021-01-29 03:22:52 -08:00
eb8d5a5e39
Integrate LLVM at llvm/llvm-project@b92a39ac13
...
Updates LLVM usage to match
[b92a39ac1319](https://github.com/llvm/llvm-project/commit/b92a39ac1319 )
PiperOrigin-RevId: 354482705
2021-01-29 00:32:48 -08:00
74b103fedf
Integrate LLVM at llvm/llvm-project@081c1db02d
...
Updates LLVM usage to match
[081c1db02dd2](https://github.com/llvm/llvm-project/commit/081c1db02dd2 )
PiperOrigin-RevId: 354470978
2021-01-28 22:34:40 -08:00
c915b698b9
Integrate LLVM at llvm/llvm-project@e29552c5af
...
Updates LLVM usage to match
[e29552c5aff6](https://github.com/llvm/llvm-project/commit/e29552c5aff6 )
PiperOrigin-RevId: 354433542
2021-01-28 17:37:31 -08:00
c97cff9a7b
[XLA:GPU] Migrate AllReduce thunk emission to MLIR.
...
PiperOrigin-RevId: 354335704
2021-01-28 10:06:06 -08:00
30ce82790d
Upstream mhlo.reduce lowering to Linalg to MHLO repo.
...
In IREE, we use indexed generic op to handle the initial value. However, we
lower it to a generic op that carries an init_tensor here, and leave the handle
of initialization problem to later passes.
PiperOrigin-RevId: 354294807
2021-01-28 05:46:09 -08:00
39589add22
Use the correct shape when converting mhlo.reshape
...
If mhlo.reshape is not purely collapsing some consecutive operand
dimensions into result dimensions, we will generate two linalg
reshape op for it: the first one collapses all operand dimensions
into one dimension, and the second one expands it to all result
dimensions. For this case, the number of collapsed/expanded dimensions
should be coming strictly from the operand/result. It is different
from the case where we can generate one linalg reshape. For that case,
the reassociation map should have rank equal to the largest among
operand/result shape.
PiperOrigin-RevId: 354293826
2021-01-28 05:37:54 -08:00
e0a7be7fb1
[MLIR][CHLO] Add `chlo.lgamma` and lowering to `hlo`
...
PiperOrigin-RevId: 354287316
2021-01-28 04:35:03 -08:00
c3ddcd6c7f
[MLIR][CHLO] Implement type inference for `is_inf`-like operations in CHLO
...
PiperOrigin-RevId: 354265834
2021-01-28 01:37:04 -08:00
fe2e5a175f
[MLIR][HLO] Implement type inference for `is_finite` op
...
PiperOrigin-RevId: 354261420
2021-01-28 00:56:12 -08:00
c653db73c5
Integrate LLVM at llvm/llvm-project@c85b6bf33c
...
Updates LLVM usage to match
[c85b6bf33c47](https://github.com/llvm/llvm-project/commit/c85b6bf33c47 )
PiperOrigin-RevId: 354136678
2021-01-27 11:46:07 -08:00
d77c9ad6fa
[MLIR][CHLO] Add `is_inf`, `is_pos_inf`, and `is_neg_inf` to CHLO dialect
...
Also add the respective lowerings to MHLO.
PiperOrigin-RevId: 354101955
2021-01-27 09:00:56 -08:00
f4f728f18e
Add SameOperandsAndResultShape trait to ConvertOp and IsFiniteOp.
...
PiperOrigin-RevId: 354070005
2021-01-27 05:10:01 -08:00
A. Unique TensorFlower
Tres Popp
Rahul Joshi
Adrian Kuegel
Stephan Herhut
Mahesh Ravishankar
Hanhan Wang
Lei Zhang