test cmakefile for cublas.

This commit is contained in:
Concedo 2023-05-15 14:50:38 +08:00
parent 923184f2e8
commit 466cd21368
4 changed files with 325 additions and 1 deletions

View file

@ -238,6 +238,7 @@ ModelLoadResult gpttype_load_model(const load_model_inputs inputs, FileFormat in
llama_ctx_params.logits_all = false;
llama_ctx_params.use_mmap = inputs.use_mmap;
llama_ctx_params.use_mlock = inputs.use_mlock;
llama_ctx_params.n_gpu_layers = inputs.gpulayers;
llama_ctx_v1 = llama_init_from_file(modelname.c_str(), llama_ctx_params);