diff --git a/paddle/pten/core/compat/arg_map_context.h b/paddle/pten/core/compat/arg_map_context.h index e7dfc0706544c9ce3f33d9e56bf406089da7f5a2..835799ec546aff468972e7adf755129950390f9f 100644 --- a/paddle/pten/core/compat/arg_map_context.h +++ b/paddle/pten/core/compat/arg_map_context.h @@ -34,6 +34,7 @@ struct KernelSignature { KernelArgsTuple args; KernelSignature() = default; + KernelSignature(std::string&& kernel_name, paddle::SmallVector&& inputs, paddle::SmallVector&& attrs, @@ -45,6 +46,14 @@ struct KernelSignature { const paddle::SmallVector& attrs, const paddle::SmallVector& 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);