llama-memory.h 850 B

123456789101112131415161718192021
  1. #pragma once
  2. #include "llama.h"
  3. // general concept of LLM memory
  4. // the KV cache is a type of LLM memory, but there can be other types
  5. class llama_memory_i {
  6. public:
  7. virtual void clear() = 0;
  8. virtual void defrag() = 0;
  9. virtual bool seq_rm (llama_seq_id seq_id, llama_pos p0, llama_pos p1) = 0;
  10. virtual void seq_cp (llama_seq_id seq_id_src, llama_seq_id seq_id_dst, llama_pos p0, llama_pos p1) = 0;
  11. virtual void seq_keep(llama_seq_id seq_id) = 0;
  12. virtual void seq_add (llama_seq_id seq_id, llama_pos p0, llama_pos p1, llama_pos delta) = 0;
  13. virtual void seq_div (llama_seq_id seq_id, llama_pos p0, llama_pos p1, int d) = 0;
  14. virtual llama_pos seq_pos_max(llama_seq_id seq_id) = 0;
  15. virtual bool get_can_edit() const = 0;
  16. };