Skip to content

Commit 3206bcd

Browse files
committed
refine log and add QPS
1 parent 06adccf commit 3206bcd

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

paddle/fluid/inference/tests/book/test_inference_nlp.cc

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -202,11 +202,13 @@ TEST(inference, nlp) {
202202
&fetch_targets, !FLAGS_prepare_vars);
203203
}
204204
stop_ms = GetCurrentMs();
205+
LOG(INFO) << "Tid: 0, process " << datasets.size()
206+
<< " samples, avg time per sample: "
207+
<< (stop_ms - start_ms) / datasets.size() << " ms";
205208
}
206209

207210
LOG(INFO) << "Total inference time with " << FLAGS_num_threads
208211
<< " threads : " << (stop_ms - start_ms) / 1000.0
209-
<< " sec, avg time per seq: "
210-
<< (stop_ms - start_ms) / datasets.size() << " ms";
212+
<< " sec, QPS: " << datasets.size() / ((stop_ms - start_ms) / 1000);
211213
delete scope;
212214
}

0 commit comments

Comments
 (0)