From fb69204f6deb4a14be903bf45c37aeac687c2e7d Mon Sep 17 00:00:00 2001 From: Huihuang Zheng Date: Sat, 4 Feb 2023 17:00:34 +0800 Subject: [PATCH] Add Some Default Parameters to CINN Interface for Country Standard (#50182) As the title --- paddle/fluid/framework/paddle2cinn/cinn_compiler.h | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/paddle/fluid/framework/paddle2cinn/cinn_compiler.h b/paddle/fluid/framework/paddle2cinn/cinn_compiler.h index b3a26781fa..50a4f99f81 100644 --- a/paddle/fluid/framework/paddle2cinn/cinn_compiler.h +++ b/paddle/fluid/framework/paddle2cinn/cinn_compiler.h @@ -22,6 +22,7 @@ #include #include +#include "cinn/common/target.h" #include "paddle/fluid/framework/ir/graph.h" #include "paddle/fluid/framework/lod_tensor.h" #include "paddle/fluid/framework/paddle2cinn/cinn_cache_key.h" @@ -75,8 +76,8 @@ class CinnCompiler { const CinnCompiledObject& Compile( const ir::Graph& graph, - const std::map& input_tensors, - const ::cinn::common::Target& target, + const std::map& input_tensors = {}, + const ::cinn::common::Target& target = ::cinn::common::DefaultTarget(), void* stream = nullptr); const CinnCompiledObject& Compile( -- GitLab