5fbdac34a9
[XLA:GPU] Add AllReduce{Start,Done} to MLIR LHLO dialect.
...
PiperOrigin-RevId: 379455720
2021-06-15 03:55:19 -07:00
14093b7906
[XLA:GPU] Add AllReduce{Start,Done} to MLIR LHLO dialect.
...
PiperOrigin-RevId: 378681070
2021-06-10 10:27:22 -07:00
968226b9d7
[XLA:GPU] Add AllReduce{Start,Done} to MLIR LHLO dialect.
...
PiperOrigin-RevId: 378640706
2021-06-10 06:54:42 -07:00
fc88cf1ff4
[HLO] Adopt custom syntax for convolution dims and window attributes for LMHLO_GPU
...
PiperOrigin-RevId: 374889917
2021-05-20 09:41:48 -07:00
6b439f7eee
[mlir][NFC] Replace usages or mlir/IR/StandardTypes.h with mlir/IR/BuiltinTypes.h
...
StandardTypes.h was moved to BuiltinTypes.h and is being removed.
PiperOrigin-RevId: 347115952
2020-12-11 19:01:25 -08:00
51cd4200b6
Make LMHLO's Dot have the same power as MHLO's DotGeneral.
...
PiperOrigin-RevId: 337391565
2020-10-15 15:09:06 -07:00
f6b4e6758a
Add GPU specific LMHLO level ops
...
- Introduce operations in a new lmhlo_gpu dialect that map to GPU library function calls
in the XLA:GPU backend.
- Add basic unit tests as well.
PiperOrigin-RevId: 337132166
2020-10-14 11:23:55 -07:00
Chris Jones
A. Unique TensorFlower
Rahul Joshi
River Riddle