[PT FE] ScaledDotProductAttention fix issue with other float dtypes (#21638)

* Fix scale input

* Improve torch scaled dot tests
This commit is contained in:
Mateusz Mikolajczyk authored and GitHub committed 2023-12-18 14:32:54 +04:00
1 parent e224e4f4f0
commit be476519fb
2 files changed
+11 -10

No files matched your search

@@ -32,7 +32,7 @@ std::shared_ptr<ov::Node> translate_scaled_dot_product_attention_common(const No
inputs.push_back(context.mark_node(std::make_shared<v1::ConvertLike>(zero, query)));
}
if (!context.input_is_none(6))
inputs.push_back(context.get_input(6));
inputs.push_back(context.mark_node(std::make_shared<v1::ConvertLike>(context.get_input(6), query)));
return context.mark_node(std::make_shared<v13::ScaledDotProductAttention>(inputs, is_causal));
}