mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-21 05:34:18 +08:00
This PR only adds the execution of the benchmarks on this PR and print results, following diffs will add checking out head~1 and running it and comparing. to access results goto test pr_time_benchmarks and inspect logs: you should see ``` + echo 'benchmark results on current PR: ' benchmark results on current PR: + cat /var/lib/jenkins/workspace/test/test-reports/pr_time_benchmarks_before.txt update_hint_regression,instruction_count,27971461254 ``` Pull Request resolved: https://github.com/pytorch/pytorch/pull/131475 Approved by: https://github.com/ezyang
28 lines
556 B
Python
28 lines
556 B
Python
import operator_benchmark as op_bench
|
|
from pt import ( # noqa: F401
|
|
qactivation_test,
|
|
qarithmetic_test,
|
|
qatembedding_ops_test,
|
|
qbatchnorm_test,
|
|
qcat_test,
|
|
qcomparators_test,
|
|
qconv_test,
|
|
qembedding_pack_test,
|
|
qembeddingbag_test,
|
|
qgroupnorm_test,
|
|
qinstancenorm_test,
|
|
qinterpolate_test,
|
|
qlayernorm_test,
|
|
qlinear_test,
|
|
qobserver_test,
|
|
qpool_test,
|
|
qrnn_test,
|
|
qtensor_method_test,
|
|
quantization_test,
|
|
qunary_test,
|
|
)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
op_bench.benchmark_runner.main()
|