From ebc7992d319dfb234de982e63c137c7b45c12e41 Mon Sep 17 00:00:00 2001 From: Alexander Belyaev Date: Tue, 8 Sep 2020 06:05:50 -0700 Subject: [PATCH] [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 --- include/mlir-hlo/Dialect/mhlo/transforms/passes.h | 2 +- lib/Dialect/mhlo/transforms/transform_unranked_hlo.cc | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/include/mlir-hlo/Dialect/mhlo/transforms/passes.h b/include/mlir-hlo/Dialect/mhlo/transforms/passes.h index 541d8e4..2f723f4 100644 --- a/include/mlir-hlo/Dialect/mhlo/transforms/passes.h +++ b/include/mlir-hlo/Dialect/mhlo/transforms/passes.h @@ -53,7 +53,7 @@ std::unique_ptr> createLegalizeToLhloPass( std::unique_ptr> createLegalizeHloToLinalgPass(); // Transforms unranked HLO operations to ranked ones where possible. -std::unique_ptr> createTransformUnrankedHloPass(); +std::unique_ptr createTransformUnrankedHloPass(); // Sinks constants implicitly captured in control flow regions. This is // necessary to export to XLA. diff --git a/lib/Dialect/mhlo/transforms/transform_unranked_hlo.cc b/lib/Dialect/mhlo/transforms/transform_unranked_hlo.cc index 58a0f0e..bfa8cf5 100644 --- a/lib/Dialect/mhlo/transforms/transform_unranked_hlo.cc +++ b/lib/Dialect/mhlo/transforms/transform_unranked_hlo.cc @@ -198,7 +198,7 @@ void PopulateTransformUnrankedHloPatterns(MLIRContext *context, // clang-format on } -std::unique_ptr<::mlir::Pass> createTransformUnrankedHloPass() { +std::unique_ptr createTransformUnrankedHloPass() { return std::make_unique(); }