From 2a172b245f3492b116cdec5e8f251c0b1f7cc941 Mon Sep 17 00:00:00 2001
From: Vasily Shamporov <vasily.shamporov@intel.com>
Date: Wed, 16 Oct 2024 16:23:02 +0200
Subject: [PATCH] Fix missing internal dimension

---
 .../intel_cpu/src/nodes/kernels/scaled_attn/executor_pa.cpp     | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/plugins/intel_cpu/src/nodes/kernels/scaled_attn/executor_pa.cpp b/src/plugins/intel_cpu/src/nodes/kernels/scaled_attn/executor_pa.cpp
index 036d715a06eb3f..9964f8b1e2303f 100644
--- a/src/plugins/intel_cpu/src/nodes/kernels/scaled_attn/executor_pa.cpp
+++ b/src/plugins/intel_cpu/src/nodes/kernels/scaled_attn/executor_pa.cpp
@@ -1566,7 +1566,7 @@ struct AttentionExecutor : public PagedAttentionExecutor {
         }
 
         if (rotated_block_indices) {
-            rotation_coefficients.assert_dims({block_size * rotated_block_indices.size(0)});
+            rotation_coefficients.assert_dims({ H * rotated_block_indices.size(0) * block_size });
         }
         output_emb.assert_dims({B_token, H * S});
         output_emb = output_emb.reshape({B_token, 1, H * S});