batched : minor fix table

This commit is contained in:
Georgi Gerganov 2023-10-08 16:35:54 +03:00
parent bf06d654de
commit 7438728d51
No known key found for this signature in database
GPG Key ID: 449E073F9DC10735

View File

@ -19,7 +19,7 @@ int main(int argc, char ** argv) {
int n_gpu_layers = 0;
std::vector<int> n_pp = { 128, 256, 512, 1024, 2048, 3584, 7680, };
std::vector<int> n_tg = { 128, 256, 512, };
std::vector<int> n_tg = { 128, 256, };
std::vector<int> n_pl = { 1, 2, 4, 8, 16, 32, };
//std::vector<int> n_pl = { 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 32, };
@ -106,7 +106,7 @@ int main(int argc, char ** argv) {
}
LOG_TEE("|%6s | %6s | %4s | %6s | %8s | %8s | %8s | %8s | %8s | %8s |\n", "PP", "TG", "B", "N_KV", "T_PP s", "S_PP t/s", "T_TG s", "S_TG t/s", "T s", "S t/s");
LOG_TEE("|%6s-+-%6s-+-%4s-+-%6s-+-%8s-+-%8s-+-%8s-+-%8s-+-%8s-+-%8s-|\n", "------", "------", "----", "--------", "--------", "--------", "--------", "--------", "--------", "--------");
LOG_TEE("|%6s-|-%6s-|-%4s-|-%6s-|-%8s-|-%8s-|-%8s-|-%8s-|-%8s-|-%8s-|\n", "------", "------", "----", "------", "--------", "--------", "--------", "--------", "--------", "--------");
for ( int i_pp = 0; i_pp < (int) n_pp.size(); ++i_pp) {
for ( int i_tg = 0; i_tg < (int) n_tg.size(); ++i_tg) {