onnxruntime
onnxruntime copied to clipboard
Allow fastgelu/skiplayernorm profile by pass args from commandline
Description: Describe your changes.
This allow us quickly launch a microbench session by, for example:
python skip_layer_norm_test.py 8 128 128 float32
Related PR: https://github.com/microsoft/onnxruntime/pull/12803 https://github.com/microsoft/onnxruntime/pull/12816 https://github.com/microsoft/onnxruntime/pull/12817 https://github.com/microsoft/onnxruntime/pull/12821 Reference: https://github.com/microsoft/onnxruntime/pull/12991 Motivation and Context
- Why is this change required? What problem does it solve?
- If it fixes an open issue, please link to the issue here.