Skip to content

Commit ab918bb

Browse files
committed
fix: Update clear signature for data argument after rebase
Branch: HybridRecurrentCache Signed-off-by: Gabe Goodhart <[email protected]>
1 parent 9ca45f3 commit ab918bb

File tree

2 files changed

+5
-4
lines changed

2 files changed

+5
-4
lines changed

src/llama-kv-cache-hybrid-recurrent.cpp

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -102,9 +102,10 @@ bool llama_kv_cache_hybrid_recurrent::get_can_shift() const {
102102
// Shifting is trivially supported for recurrent
103103
return kv_attn->get_can_shift();
104104
}
105-
void llama_kv_cache_hybrid_recurrent::clear() {
106-
kv_attn ->clear();
107-
kv_recurrent->clear();
105+
106+
void llama_kv_cache_hybrid_recurrent::clear(bool data) {
107+
kv_attn ->clear(data);
108+
kv_recurrent->clear(data);
108109
}
109110

110111
bool llama_kv_cache_hybrid_recurrent::seq_rm(llama_seq_id seq_id, llama_pos p0, llama_pos p1) {

src/llama-kv-cache-hybrid-recurrent.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -55,7 +55,7 @@ class llama_kv_cache_hybrid_recurrent : public llama_memory_i {
5555

5656
bool get_can_shift() const override;
5757

58-
void clear() override;
58+
void clear(bool data) override;
5959

6060
bool seq_rm (llama_seq_id seq_id, llama_pos p0, llama_pos p1) override;
6161
void seq_cp (llama_seq_id seq_id_src, llama_seq_id seq_id_dst, llama_pos p0, llama_pos p1) override;

0 commit comments

Comments
 (0)