Fix

Author: vshampor

src/core/tests/reference/xattention.cpp

@@ -520,19 +520,17 @@ std::vector<BlockSelectTestData> BLOCK_SELECT_TEST_CASES = {
         {2, 2, 5},
         // clang-format off
         {
-            // values in first blocks have no impact on attention mass
-            // since they are always selected
             0.0000, 0.5151, 0.4323, 0.5014, 0.5513,
-            -100.0, 0.4557, 0.4876, 0.5870, 0.4697,
+            100.0, 0.4557, 0.4876, 0.5870, 0.4697,
 
             1.7491, 0.3118, 0.4507, 0.5180, 0.5194,
             0.3123, 0.5315, 0.4446, 0.4929, 0.5310
         },
      // clang-format on
      /* threshold = */ 0.25,
      {
-         {{0, 0}, {1, 0}, {0, 3}, {1, 4}, {0, 1}, {1, 3}},
-         {{0, 0}, {1, 0}, {0, 3}, {1, 4}, {0, 2}, {1, 1}},
+         {{0, 0}, {1, 0}, {0, 3}, {1, 4}},
+         {{0, 0}, {1, 0}, {0, 3}, {1, 4}},
      }},
 
     {{2, 2, 5},
@@ -548,8 +546,8 @@ std::vector<BlockSelectTestData> BLOCK_SELECT_TEST_CASES = {
      // clang-format on
      /* threshold = */ 0.45,
      {
-         {{0, 0}, {1, 0}, {0, 3}, {1, 4}, {0, 1}, {1, 2}, {1, 3}},
-         {{0, 0}, {1, 0}, {0, 3}, {1, 4}, {0, 1}, {1, 1}, {1, 3}},
+         {{0, 0}, {1, 0}, {0, 3}, {1, 4}, {1, 3}},
+         {{0, 0}, {1, 0}, {0, 3}, {1, 4}, {1, 1}},
      }},
     {{2, 2, 5},
      // clang-format off
@@ -558,13 +556,13 @@ std::vector<BlockSelectTestData> BLOCK_SELECT_TEST_CASES = {
             0.5267, 0.4557, 0.4876, 0.5870, 0.4697,
 
             0.4647, 0.5118, 0.4507, 0.5180, 0.5194,
-            0.0000, 0.1234, 0.4446, 0.4929, 0.5310
+            0.0000, 0.4446, 0.1234, 0.4929, 0.5310
         },
      // clang-format on
      /* threshold = */ 0.8,
      {
-         {{0, 0}, {1, 0}, {0, 3}, {1, 4}, {0, 1}, {0, 2}, {1, 1}, {1, 2}, {1, 3}, {1, 4}},
-         {{0, 0}, {1, 0}, {0, 3}, {1, 4}, {0, 1}, {0, 2}, {1, 2}, {1, 3}},
+         {{0, 0}, {1, 0}, {0, 3}, {1, 4}, {0, 1}, {0, 2}, {1, 3}, {1, 2}},
+         {{0, 0}, {1, 0}, {0, 3}, {1, 4}, {0, 1}, {0, 2}, {1, 3}, {1, 1}},
      }},
     {{2, 2, 5},
      // clang-format off