GithubStar/ncnn
1
0
Fork 0
mirror of https://github.com/Tencent/ncnn.git synced 2026-10-24 17:50:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

pnnx drop sdap scale=None for compatiblity with old torch (#5107)

Browse Source
This commit is contained in:
nihui
2023-10-27 11:50:52 +08:00
committed by GitHub
parent 14e14a9ae8
commit 3116e028d2
1 changed files with 11 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tools/pnnx/src/pass_level2/F_scaled_dot_product_attention.cpp
+11
View File
@@ -65,6 +65,17 @@ pnnx.Output output 1 0 out
{
return "F.scaled_dot_product_attention";
}
void write(Operator* op, const std::map<std::string, Parameter>& captured_params, const std::map<std::string, Attribute>& captured_attrs) const
{
GraphRewriterPass::write(op, captured_params, captured_attrs);
if (captured_params.at("scale").type == 0)
{
// drop scale=None for compatiblity with old torch
op->params.erase("scale");
}
}
};
REGISTER_GLOBAL_PNNX_GRAPH_REWRITER_PASS(F_scaled_dot_product_attention_1, 10)