Skip to content

Commit cb91ec8

Browse files
kouarthw
authored andcommitted
llama : fix llama_split_mode enum values in main_gpu document (ggml-org#9057)
LLAMA_SPLIT_* were renamed to LLAMA_SPLIT_MODE_* in ggml-org#5697.
1 parent c2e3b35 commit cb91ec8

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

include/llama.h

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -267,9 +267,9 @@ extern "C" {
267267
enum llama_split_mode split_mode; // how to split the model across multiple GPUs
268268

269269
// main_gpu interpretation depends on split_mode:
270-
// LLAMA_SPLIT_NONE: the GPU that is used for the entire model
271-
// LLAMA_SPLIT_ROW: the GPU that is used for small tensors and intermediate results
272-
// LLAMA_SPLIT_LAYER: ignored
270+
// LLAMA_SPLIT_MODE_NONE: the GPU that is used for the entire model
271+
// LLAMA_SPLIT_MODE_ROW: the GPU that is used for small tensors and intermediate results
272+
// LLAMA_SPLIT_MODE_LAYER: ignored
273273
int32_t main_gpu;
274274

275275
// proportion of the model (layers or rows) to offload to each GPU, size: llama_max_devices()

0 commit comments

Comments
 (0)