Skip to content

Commit b40e4f2

Browse files
tdakhranggerganov
andcommitted
Update src/llama-memory-hybrid-iswa.cpp [no-ci]
Co-authored-by: Georgi Gerganov <ggerganov@gmail.com>
1 parent 88c90a7 commit b40e4f2

1 file changed

Lines changed: 2 additions & 6 deletions

File tree

src/llama-memory-hybrid-iswa.cpp

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -183,16 +183,12 @@ std::map<ggml_backend_buffer_type_t, size_t> llama_memory_hybrid_iswa::memory_br
183183
}
184184

185185
void llama_memory_hybrid_iswa::state_write(llama_io_write_i & io, llama_seq_id seq_id, llama_state_seq_flags flags) const {
186-
if ((flags & LLAMA_STATE_SEQ_FLAGS_PARTIAL_ONLY) == 0) {
187-
mem_attn->state_write(io, seq_id, flags);
188-
}
186+
mem_attn->state_write(io, seq_id, flags);
189187
mem_recr->state_write(io, seq_id, flags);
190188
}
191189

192190
void llama_memory_hybrid_iswa::state_read(llama_io_read_i & io, llama_seq_id seq_id, llama_state_seq_flags flags) {
193-
if ((flags & LLAMA_STATE_SEQ_FLAGS_PARTIAL_ONLY) == 0) {
194-
mem_attn->state_read(io, seq_id, flags);
195-
}
191+
mem_attn->state_read(io, seq_id, flags);
196192
mem_recr->state_read(io, seq_id, flags);
197193
}
198194

0 commit comments

Comments
 (0)