From 613f8bf75f8c4f7a0101898c14394e7290ff524d Mon Sep 17 00:00:00 2001
From: Vasily Shamporov <vasily.shamporov@intel.com>
Date: Thu, 17 Oct 2024 11:31:34 +0200
Subject: [PATCH] Fix rotation function signature

---
 src/cpp/src/model_runner.hpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/cpp/src/model_runner.hpp b/src/cpp/src/model_runner.hpp
index 68254844f4..bf06c9e315 100644
--- a/src/cpp/src/model_runner.hpp
+++ b/src/cpp/src/model_runner.hpp
@@ -173,7 +173,7 @@ class ModelRunner {
         _set_block_indices(sequence_groups, scheduler_output, total_num_blocks);
 
         if (!m_cache_rotation_coefficients.empty()) {
-            _set_cache_rotation_coefficients();
+            _set_cache_rotation_coefficients(sequence_groups, scheduler_output);
         }
 
         m_request.set_tensor("block_indices_begins", block_indices_begins);