Skip to content

Commit

Permalink
fixes
Browse files Browse the repository at this point in the history
  • Loading branch information
aobolensk committed Feb 14, 2025
1 parent c65b672 commit 0e782b7
Show file tree
Hide file tree
Showing 2 changed files with 8 additions and 7 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -187,13 +187,12 @@ TEST_F(TransformationTestsF, ScaledDotProductAttentionDecompositionDynamic) {
}
}

const std::shared_ptr<ov::Node> scaled_dot_product_attention_decomposition(
std::shared_ptr<ov::Node> query,
std::shared_ptr<ov::Node> key,
std::shared_ptr<ov::Node> value,
std::shared_ptr<ov::Node> attention_mask,
std::shared_ptr<ov::Node> scale,
bool casual) {
const std::shared_ptr<ov::Node> scaled_dot_product_attention_decomposition(std::shared_ptr<ov::Node> query,
std::shared_ptr<ov::Node> key,
std::shared_ptr<ov::Node> value,
std::shared_ptr<ov::Node> attention_mask,
std::shared_ptr<ov::Node> scale,
bool casual) {
const auto q_shape = std::make_shared<ov::op::v3::ShapeOf>(query, element::i32);
const auto k_shape = std::make_shared<ov::op::v3::ShapeOf>(key, element::i32);
const auto minus_one = ov::op::v0::Constant::create(element::i32, Shape{}, {-1});
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -134,9 +134,11 @@ static bool is_segfault_detector_emitter(const intel_cpu::jit_emitter* emitter)
#define CREATE_CPU_EMITTER(e_type) \
{ \
[this](const snippets::lowered::ExpressionPtr& expr) -> std::shared_ptr<snippets::Emitter> { \
(void) expr; \
return std::make_shared<e_type>(h.get(), isa, expr->get_node()); \
}, \
[](const std::shared_ptr<ov::Node>& n) -> std::set<std::vector<element::Type>> { \
(void) n; \
return e_type::get_supported_precisions(n); \
} \
}
Expand Down

0 comments on commit 0e782b7

Please sign in to comment.