File tree 1 file changed +2
-2
lines changed
1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -1936,7 +1936,7 @@ int32_t llama_kv_cache_recurrent::s_copy(int i) const {
1936
1936
1937
1937
// ////////////////////////////////////////////
1938
1938
// TODO: this should not mutate the KV cache !
1939
- kv_cell & cell = const_cast <kv_cell &>(cells[i ]);
1939
+ kv_cell & cell = const_cast <kv_cell &>(cells[cell_id ]);
1940
1940
1941
1941
// prevent out-of-bound sources
1942
1942
if (cell.src < 0 || (uint32_t ) cell.src >= size) {
@@ -1959,7 +1959,7 @@ float llama_kv_cache_recurrent::s_mask(int i) const {
1959
1959
1960
1960
// ////////////////////////////////////////////
1961
1961
// TODO: this should not mutate the KV cache !
1962
- kv_cell & cell = const_cast <kv_cell &>(cells[i ]);
1962
+ kv_cell & cell = const_cast <kv_cell &>(cells[cell_id ]);
1963
1963
1964
1964
float res = (float ) (cell.src >= 0 );
1965
1965
You can’t perform that action at this time.
0 commit comments