Skip to content

Commit 3ad6b9c

Browse files
committed
fix tests
Signed-off-by: fishbell <bell.song@intel.com> fix typo Signed-off-by: fishbell <bell.song@intel.com>
1 parent 4e155ca commit 3ad6b9c

File tree

3 files changed

+2
-5
lines changed

3 files changed

+2
-5
lines changed

src/bindings/python/src/pyopenvino/core/offline_transformations.cpp

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -158,7 +158,6 @@ void regmodule_offline_transformations(py::module m) {
158158
allow_score_aggregation,
159159
allow_cache_rotation,
160160
allow_xattention,
161-
allow_adaptive_rkv);
162161
allow_adaptive_rkv,
163162
allow_qq_bias);
164163
manager.run_passes(model);
@@ -170,7 +169,7 @@ void regmodule_offline_transformations(py::module m) {
170169
py::arg("allow_cache_rotation") = false,
171170
py::arg("allow_xattention") = false,
172171
py::arg("allow_adaptive_rkv") = false,
173-
py::arg("allow_qq_bias") = false);
172+
py::arg("allow_qq_bias") = false);
174173

175174
m_offline_transformations.def(
176175
"stateful_to_stateless_transformation",

src/plugins/intel_cpu/src/nodes/paged_attn.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -99,7 +99,7 @@ void PagedAttention::initSupportedPrimitiveDescriptors() {
9999
creatorsMap.at(LayoutType::ncsp)
100100
->createSharedDesc(rtPrecision, getInputShapeAtPort(PagedAttentionExecutor::ID_V)));
101101

102-
CPU_NODE_ASSERT(orgInputNumber == 27U, "The input number of PagedAttention should be 29.");
102+
CPU_NODE_ASSERT(orgInputNumber == 27U, "The input number of PagedAttention should be 27.");
103103
// kvcache, float, []
104104
auto past_key_input_mem_precision = getOriginalInputPrecisionAtPort(PagedAttentionExecutor::ID_KCACHE);
105105
auto past_value_input_mem_precision = getOriginalInputPrecisionAtPort(PagedAttentionExecutor::ID_VCACHE);

tests/model_hub_tests/transformation_tests/test_pa_transformation.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -95,8 +95,6 @@ def apply_transformation_and_compare_diffs(ov_model: ov.Model,
9595
if (allow_qq_bias):
9696
interesting_input_patterns["qq_bias"] = r'^qq_bias$';
9797
interesting_input_patterns["qq_bias_begins"] = r'^qq_bias_begins$';
98-
interesting_input_patterns["block_update_indices"] = r'^block_update_indices$';
99-
interesting_input_patterns["block_update_indices_begins"] = r'^block_update_indices_begins$';
10098

10199
input_counters = {k: 0 for k in interesting_input_patterns}
102100
output_counters = {k: 0 for k in interesting_output_patterns}

0 commit comments

Comments
 (0)