未验证 提交 fb69204f 编写于 作者: H Huihuang Zheng 提交者: GitHub

Add Some Default Parameters to CINN Interface for Country Standard (#50182)

As the title
上级 e9629964
...@@ -22,6 +22,7 @@ ...@@ -22,6 +22,7 @@
#include <string> #include <string>
#include <unordered_map> #include <unordered_map>
#include "cinn/common/target.h"
#include "paddle/fluid/framework/ir/graph.h" #include "paddle/fluid/framework/ir/graph.h"
#include "paddle/fluid/framework/lod_tensor.h" #include "paddle/fluid/framework/lod_tensor.h"
#include "paddle/fluid/framework/paddle2cinn/cinn_cache_key.h" #include "paddle/fluid/framework/paddle2cinn/cinn_cache_key.h"
...@@ -75,8 +76,8 @@ class CinnCompiler { ...@@ -75,8 +76,8 @@ class CinnCompiler {
const CinnCompiledObject& Compile( const CinnCompiledObject& Compile(
const ir::Graph& graph, const ir::Graph& graph,
const std::map<std::string, const phi::DenseTensor*>& input_tensors, const std::map<std::string, const phi::DenseTensor*>& input_tensors = {},
const ::cinn::common::Target& target, const ::cinn::common::Target& target = ::cinn::common::DefaultTarget(),
void* stream = nullptr); void* stream = nullptr);
const CinnCompiledObject& Compile( const CinnCompiledObject& Compile(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册