Skip to content

Commit

Permalink
code style
Browse files Browse the repository at this point in the history
Signed-off-by: Vladimir Paramuzov <[email protected]>
  • Loading branch information
vladimir-paramuzov committed Dec 23, 2024
1 parent b5455a8 commit 5f452cd
Show file tree
Hide file tree
Showing 3 changed files with 5 additions and 3 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -52,7 +52,7 @@ namespace pass {
/// └─────────┘
class TRANSFORMATIONS_API SDPAFusion : public ov::pass::MatcherPass {
public:
OPENVINO_RTTI("SDPAFusion", "0");
OPENVINO_MATCHER_PASS_RTTI("SDPAFusion", "0");
SDPAFusion();
};

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -50,7 +50,7 @@ namespace pass {
/// 2. Q_scale * SDPA_Scale == 1 or K_scale * SDPA_Scale == 1
class TRANSFORMATIONS_API SDPAScaleFusion : public ov::pass::MatcherPass {
public:
OPENVINO_RTTI("SDPAScaleFusion", "0");
OPENVINO_MATCHER_PASS_RTTI("SDPAScaleFusion", "0");
SDPAScaleFusion();
};

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,7 @@
//

#include "transformations/common_optimizations/sdpa_fusion.hpp"

#include "openvino/core/rt_info.hpp"
#include "openvino/core/type.hpp"
#include "openvino/op/add.hpp"
Expand Down Expand Up @@ -30,7 +31,8 @@ SDPAFusion::SDPAFusion() {
auto mask = makePattern();

auto k_transpose_order = pattern::wrap_type<ov::op::v0::Constant>([](const Output<Node>& node) {
auto axis_order = std::dynamic_pointer_cast<ov::op::v0::Constant>(node.get_node_shared_ptr())->cast_vector<int64_t>();
auto axis_order =
std::dynamic_pointer_cast<ov::op::v0::Constant>(node.get_node_shared_ptr())->cast_vector<int64_t>();
return axis_order == std::vector<int64_t>{0, 1, 3, 2};
});

Expand Down

0 comments on commit 5f452cd

Please sign in to comment.