Browse Source

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

tags/20231027
nihui GitHub 2 years ago
parent
commit
3116e028d2
No known key found for this signature in database GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 11 additions and 0 deletions
  1. +11
    -0
      tools/pnnx/src/pass_level2/F_scaled_dot_product_attention.cpp

+ 11
- 0
tools/pnnx/src/pass_level2/F_scaled_dot_product_attention.cpp 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)


Loading…
Cancel
Save