diff --git a/paddle/fluid/framework/paddle2cinn/cinn_compiler.h b/paddle/fluid/framework/paddle2cinn/cinn_compiler.h index b3a26781fac811efc767af9744a7eb43ea0b1017..50a4f99f81c0bb4c8f050c5234b45f5ba15fac91 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(