fixed compile error: GGML_BACKEND_TYPE_GPU (+1 squashed commits)

Squashed commits:

[00ca282a] fixed compile error: LLAMA_SPLIT_MODE_ROW
This commit is contained in:
Concedo
2024-02-26 10:49:02 +08:00
parent 3ccaf8e09a
commit d47e13c892
2 changed files with 3 additions and 3 deletions
+2 -2
View File
@@ -970,9 +970,9 @@ ModelLoadResult gpttype_load_model(const load_model_inputs inputs, FileFormat in
model_params.main_gpu = cu_parseinfo_maindevice;
#if defined(GGML_USE_CUBLAS)
model_params.split_mode = (inputs.use_rowsplit?llama_split_mode::LLAMA_SPLIT_ROW:llama_split_mode::LLAMA_SPLIT_LAYER);
model_params.split_mode = (inputs.use_rowsplit?llama_split_mode::LLAMA_SPLIT_MODE_ROW:llama_split_mode::LLAMA_SPLIT_MODE_LAYER);
#else
model_params.split_mode = llama_split_mode::LLAMA_SPLIT_LAYER;
model_params.split_mode = llama_split_mode::LLAMA_SPLIT_MODE_LAYER;
#endif
llama_ctx_params.n_batch = kcpp_params->n_batch;