frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

add new ops to benchmark_all_test (#21365)

Browse Source 
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/21365

This diff adds new operators to benchmark_all_test so all the supported ops can be built as one binary

Reviewed By: hl475

Differential Revision: D15627328

fbshipit-source-id: b7ca550a279f485102a6a6bd47e4032c7beb9940
This commit is contained in:
Mingzhe Li
2019-06-04 13:51:16 -07:00
committed by Facebook Github Bot
parent 2ed6f017ed
commit b869a3b4ac
3 changed files with 8 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
benchmarks/operator_benchmark/ops/benchmark_all_test.py
View File

@ -3,11 +3,14 @@ from __future__ import division
from __future__ import print_function
from __future__ import unicode_literals
from operator_benchmark import benchmark_runner
from operator_benchmark.ops import ( # noqa
add_test, # noqa
matmul_test) # noqa
import operator_benchmark as op_bench
from ops.pt import ( # noqa
add_test, batchnorm_test, conv_test, linear_test, matmul_test # noqa
)
from ops.c2 import ( # noqa
add_test, matmul_test # noqa
)
if __name__ == "__main__":
benchmark_runner.main()
op_bench.benchmark_runner.main()

0
benchmarks/operator_benchmark/ops/c2/__init__.py Normal file
View File

0
benchmarks/operator_benchmark/ops/pt/__init__.py Normal file
View File