colin
/
mlir-hlo
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[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
This commit is contained in:
Alexander Belyaev 2020-09-08 06:05:50 -07:00 committed by TensorFlow MLIR Team
parent 64d9a465ee
commit ebc7992d31
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
include/mlir-hlo/Dialect/mhlo/transforms/passes.h
View File

@ -53,7 +53,7 @@ std::unique_ptr<OperationPass<ModuleOp>> createLegalizeToLhloPass(
std::unique_ptr<OperationPass<FuncOp>> createLegalizeHloToLinalgPass(); std::unique_ptr<OperationPass<FuncOp>> createLegalizeHloToLinalgPass();
// Transforms unranked HLO operations to ranked ones where possible. // Transforms unranked HLO operations to ranked ones where possible.
std::unique_ptr<OperationPass<FuncOp>> createTransformUnrankedHloPass(); std::unique_ptr<FunctionPass> createTransformUnrankedHloPass();
// Sinks constants implicitly captured in control flow regions. This is // Sinks constants implicitly captured in control flow regions. This is
// necessary to export to XLA. // necessary to export to XLA.

2
lib/Dialect/mhlo/transforms/transform_unranked_hlo.cc
View File

@ -198,7 +198,7 @@ void PopulateTransformUnrankedHloPatterns(MLIRContext *context,
// clang-format on // clang-format on
} }
std::unique_ptr<::mlir::Pass> createTransformUnrankedHloPass() { std::unique_ptr<FunctionPass> createTransformUnrankedHloPass() {
return std::make_unique<TransformUnrankedHloPass>(); return std::make_unique<TransformUnrankedHloPass>();
} }