Commit graph

7 commits

Author SHA1 Message Date
Justin Chu
fdce4fa6af
Format all python files under onnxruntime with black and isort (#11324)
Description: Format all python files under onnxruntime with black and isort.

After checking in, we can use .git-blame-ignore-revs to ignore the formatting PR in git blame.

#11315, #11316
2022-04-26 09:35:16 -07:00
PeixuanZuo
5763657715
[UPDATE] Add prefix in front of the file (#10884) 2022-03-16 21:05:18 +08:00
PeixuanZuo
040c0645e2
[ADD] Add micro-benchmark for Cast (#10870)
* [ADD] Add micro-benchmark for Cast

* [UPDATE] related to bert model and fix the format
2022-03-16 10:48:26 +08:00
zhangyaobit
9cbcc93e03
Add micro-benchmarks for Attention and SkipLayerNormalization ops. (#10798)
* Add micro-benchmarks for Attention and SkipLayerNormalization ops.

* Add choices for argument provider and precision.

* Automatically select CUDA or ROCM execution provider.
2022-03-09 18:18:51 -08:00
zhangyaobit
b7f00b9682
Refactor the common code per operator into an abstract base class. (#10785) 2022-03-07 13:15:49 -08:00
zhangyaobit
4c88fa5971
Add micro-benchmark for FastGelu (#10744)
* Add micro-benchmark for FastGelu

* Delete the bert-base case, as it is very similar to the bert-large one.

* Add argument parsing and more user-friendly provider type assertion.
2022-03-04 08:51:15 -08:00
zhangyaobit
a7738b52c5
Add microbench to benchmark single operators. (#10678)
* Add microbench to benchmark single operators.

* Move to tool directory; seperate data genration from io binding.

* Refector.

* Clean up.

* Use precision instead for extensibility.

* Refactor the create_io_binding function to take in torch tensors
instead of numpy arrays; this reflects more accurately what
the function does, because it is torch tensors that got bound.
2022-03-01 16:00:16 -08:00