未验证 提交 33e3f5ac 编写于 作者: C Chen Weihang 提交者: GitHub

Add kernelsignature constructor for windows (#39253)

* add constructor for win

* change impl

* fix bug
上级 98c1829b
...@@ -34,6 +34,7 @@ struct KernelSignature { ...@@ -34,6 +34,7 @@ struct KernelSignature {
KernelArgsTuple args; KernelArgsTuple args;
KernelSignature() = default; KernelSignature() = default;
KernelSignature(std::string&& kernel_name, KernelSignature(std::string&& kernel_name,
paddle::SmallVector<std::string>&& inputs, paddle::SmallVector<std::string>&& inputs,
paddle::SmallVector<std::string>&& attrs, paddle::SmallVector<std::string>&& attrs,
...@@ -45,6 +46,14 @@ struct KernelSignature { ...@@ -45,6 +46,14 @@ struct KernelSignature {
const paddle::SmallVector<std::string>& attrs, const paddle::SmallVector<std::string>& attrs,
const paddle::SmallVector<std::string>& outputs) const paddle::SmallVector<std::string>& outputs)
: name(kernel_name), args(std::make_tuple(inputs, attrs, outputs)) {} : name(kernel_name), args(std::make_tuple(inputs, attrs, outputs)) {}
// TODO(chenweihang): add assign constructor to solve windows compile
// problem, remove it later
KernelSignature& operator=(const KernelSignature& other) {
name = other.name;
args = other.args;
return *this;
}
}; };
std::ostream& operator<<(std::ostream& os, KernelSignature signature); std::ostream& operator<<(std::ostream& os, KernelSignature signature);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册