458e861254
[MLIR][KerneGen] Lower `tf.Atan` all the way to LLVM
...
PiperOrigin-RevId: 334810730
2020-10-01 07:38:40 -07:00
4002077261
[MLIR][KernelGen] Lower `tf.Sinh` to MLHLO
...
PiperOrigin-RevId: 332425724
2020-09-18 04:27:07 -07:00
2fbbbe9cf1
[MLIR][KernelGen] Lower `tf.Acos` to LMHLO.
...
- Add ranked code generation for `mhlo.compare/select`
- Add bufferization for `tensor_cast`
- Add lowerings for `Atan2Op`
PiperOrigin-RevId: 332407734
2020-09-18 01:40:18 -07:00
69b80d8deb
[MLIR] Extend unranked transformation to CHLO dialect
...
PiperOrigin-RevId: 332026604
2020-09-16 09:49:18 -07:00
da43c8596b
[MLIR] Simplify and generalize `transform-unranked-hlo`
...
This refactoring allows to support a wider range of n-ary operations in future
changes.
PiperOrigin-RevId: 331953362
2020-09-16 01:13:23 -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
36ddbeb6b2
Remove the dependency on global dialect registry from mlir-hlo
...
PiperOrigin-RevId: 328457105
2020-08-25 20:30:42 -07:00
843af36e05
[MLIR] Add e2e test for unranked unary TF op, lowered and run with CPU runner.
...
PiperOrigin-RevId: 325665428
2020-08-09 02:38:00 -07:00
5d3cc2105e
[MLIR][HLO] Remove redundant casts from unranked to ranked transformation
...
The transformation of unranked to ranked operations no longer generates cast
operations for shapes and sizes. Instead, we use the newly introduced support
for extent tensor and index types directly.
PiperOrigin-RevId: 325057440
2020-08-05 11:11:43 -07:00
cd01bb4c4e
More cleanup in mlir-hlo to prepare for the standalone build
...
Shuffle files around, use TableGen to register passes, and introduce
a `mlir-hlo-opt.cpp` file to hold the main entry point of the -opt tool
and stop relying on static registration for dialect/passes.
PiperOrigin-RevId: 323674455
2020-08-03 19:28:00 -07:00
effd3fb4f9
Extend unranked to ranked pattern for hlo operations to all unary and binary ops.
...
As this is essentially always the same pattern, only one operation is tested.
PiperOrigin-RevId: 323525418
2020-08-03 19:27:49 -07:00
739758f9cc
Integrate LLVM at llvm/llvm-project@eed333149d
...
Updates LLVM usage to match
[eed333149d17](https://github.com/llvm/llvm-project/commit/eed333149d17 )
PiperOrigin-RevId: 323354988
2020-08-03 19:27:25 -07:00
506ddd9c4a
Cleanup build rule names in compiler/mlir/hlo to remove the redundant/obsolete xla_ prefix
...
PiperOrigin-RevId: 320320140
2020-07-30 22:33:29 +00:00
A. Unique TensorFlower
Alexander Belyaev
Mehdi Amini
Stephan Herhut
Thomas Joerg