Skip to content

Commit 4fcee65

Browse files
timxu826e-ddykim
authored andcommitted
fix type issue.
1 parent 7f954bb commit 4fcee65

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/common/transformations/src/transformations/common_optimizations/sdpa_scale_fusion.cpp

+2-2
Original file line numberDiff line numberDiff line change
@@ -83,11 +83,11 @@ SDPAScaleFusion::SDPAScaleFusion() {
8383
// Extract scalar scale values for Q and K if those are constant and set new inputs for SDPA
8484
if (has_q_scale) {
8585
scale_q_node = pattern_map.at(scale_q).get_node_shared_ptr();
86-
if(pattern_map.at(q).get_element_type() == q_input.get_element_type()) {
86+
if (pattern_map.at(q).get_element_type() == q_input.get_element_type()) {
8787
if (ov::is_type<ov::op::v0::Constant>(scale_q_node)) {
8888
scale_q_value = ov::as_type_ptr<ov::op::v0::Constant>(scale_q_node)->cast_vector<float>()[0];
8989
q_input = pattern_map.at(q);
90-
}
90+
}
9191
} else {
9292
has_q_scale = false;
9393
}

0 commit comments

Comments
 (0)