Add an option to set number of warmup iterations#27739
Add an option to set number of warmup iterations#27739luotao1 merged 1 commit intoPaddlePaddle:developfrom
Conversation
|
Thanks for your contribution! |
luotao1
left a comment
There was a problem hiding this comment.
Does this PR make the elapsed time of the unit-test become longer since each unit-test has an extra warmup iteration? Could the default warmup_iter be 0?
@luotao1 |
|
Got it, thanks for your explanation. |
PR types
Others
PR changes
Others
Describe
With this patch one can set the number of iterations for warmup phase in performance tests. When the option
--warmup=trueis used, by default one iteration of inference is run as a warmup. Now, with additional option--warmup_iters=5the given number of iterations will be run in warmup.This option helps properly measure performance of GRU FP32/INT8 model with oneDNN. When running GRU inference with oneDNN, the oneDNN cache needs more than one (actually close to 20) iterations to cache all the primitives required during inference of the whole dataset.