vrr/prima.cpp
2
0
Fork 0
mirror of https://github.com/Lizonghang/prima.cpp.git synced 2025-09-10 11:44:33 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

add api llama_model_set_n_gpu_layers

Browse source
This commit is contained in:
Lizonghang 2025-01-15 10:47:53 +04:00
parent 9279a2e3ff
commit 1e1ba5bb91
2 changed files with 9 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

1
common/common.cpp
View file

@ -1440,6 +1440,7 @@ struct llama_init_result llama_init_from_gpt_params(gpt_params & params) {
params.n_gpu_layers = n_gpu_layers[my_rank];
cparams.n_gpu_layers = n_gpu_layers[my_rank];
mparams.n_gpu_layers = n_gpu_layers[my_rank];
llama_model_set_n_gpu_layers(model, n_gpu_layers[my_rank]);
#ifdef LLAMA_DEBUG
device_print_props(dev_info_set, n_world, model, cparams);