Update profiler tool to support gpt2 and longformer models (microsoft…

…#6011) * support gpt2 and longformer in profiler tool * rename bert_profiler to profiler * Add --basic_optimization to allow user to use basic level of graph optimization * Add --kernel_time_only to filter kernel time and exclude fence time * Add --threshold to filter nodes that with low run time percentage.
jjsbear · Dec 7, 2020 · 51fbe87 · 51fbe87
1 parent 925879a
commit 51fbe87
Show file tree

Hide file tree

Showing 2 changed files with 370 additions and 188 deletions.
diff --git a/onnxruntime/python/tools/transformers/bert_profiler.py b/onnxruntime/python/tools/transformers/bert_profiler.py