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

show device props

Browse source
This commit is contained in:
Lizonghang 2025-01-18 17:25:27 +04:00
parent f9d16fbf71
commit 6761ca5358
2 changed files with 10 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

19
common/profiler.cpp
View file

@ -1966,17 +1966,16 @@ void device_print_props(struct device_info * dev_info_set, int n, struct llama_m
LOG_INF("| %-10" PRId64 " ", dev_info_set[0].model_bytes.nb_output);
LOG_INF("\n");
// todo: calculate for each device, not only master
float latency = 0.0f;
int n_layers = llama_model_n_layers (model);
latency += device_compute_delay (dev_info_set[0], n_layers, cparams);
latency += device_memory_access_delay(dev_info_set[0], model, cparams, n_layers);
latency += device_disk_access_delay (dev_info_set[0], model, cparams); // if physical memory is not enough, some mapped data will be released and reloaded later
latency += device_mem_copy_delay (dev_info_set[0], model, cparams); // memory copy delay in kvcache
// float latency = 0.0f;
// int n_layers = llama_model_n_layers (model);
// latency += device_compute_delay (dev_info_set[0], n_layers, cparams);
// latency += device_memory_access_delay(dev_info_set[0], model, cparams, n_layers);
// latency += device_disk_access_delay (dev_info_set[0], model, cparams); // if physical memory is not enough, some mapped data will be released and reloaded later
// latency += device_mem_copy_delay (dev_info_set[0], model, cparams); // memory copy delay in kvcache
LOG_INF("| Token latency (ms) ");
LOG_INF("| %-10.2f ", latency);
LOG_INF("\n");
// LOG_INF("| Token latency (ms) ");
// LOG_INF("| %-10.2f ", latency);
// LOG_INF("\n");
LOG_INF("-------------------------------------------------------------------------------------------\n\n");
}