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

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

As the title
上级 e9629964
......@@ -22,6 +22,7 @@
#include <string>
#include <unordered_map>
#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<std::string, const phi::DenseTensor*>& input_tensors,
const ::cinn::common::Target& target,
const std::map<std::string, const phi::DenseTensor*>& input_tensors = {},
const ::cinn::common::Target& target = ::cinn::common::DefaultTarget(),
void* stream = nullptr);
const CinnCompiledObject& Compile(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册