llama : document that only one on-device state can be saved per sequence (#23520)
This commit is contained in:
+2
-1
@@ -874,7 +874,8 @@ extern "C" {
|
|||||||
// work only with partial states, such as SWA KV cache or recurrent cache (e.g. Mamba)
|
// work only with partial states, such as SWA KV cache or recurrent cache (e.g. Mamba)
|
||||||
#define LLAMA_STATE_SEQ_FLAGS_PARTIAL_ONLY 1
|
#define LLAMA_STATE_SEQ_FLAGS_PARTIAL_ONLY 1
|
||||||
|
|
||||||
// keeps the tensor data on device buffers (i.e. not accessible in host memory, but faster save/load)
|
// Keeps the tensor data on device buffers (i.e. not accessible in host memory, but faster save/load).
|
||||||
|
// Getting the state for a seq_id with this flag invalidates all prior states gotten for that seq_id with this flag.
|
||||||
#define LLAMA_STATE_SEQ_FLAGS_ON_DEVICE 2
|
#define LLAMA_STATE_SEQ_FLAGS_ON_DEVICE 2
|
||||||
|
|
||||||
typedef uint32_t llama_state_seq_flags;
|
typedef uint32_t llama_state_seq_flags;
|
||||||
|
|||||||
Reference in New Issue
Block a user