mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/30048 as title (Note: this ignores all push blocking failures!) Test Plan: ``` buck run mode/opt caffe2/benchmarks/operator_benchmark:benchmark_all_other_test # ---------------------------------------- # PyTorch/Caffe2 Operator Micro-benchmarks # ---------------------------------------- # Tag : short # Benchmarking PyTorch: add # Mode: Eager # Name: add_M64_N64_K64_cpu # Input: M: 64, N: 64, K: 64, device: cpu Forward Execution Time (us) : 142.032 ... Reviewed By: hl475 Differential Revision: D18580754 fbshipit-source-id: 125482d2987cbdb1d019ccedf56a9da5a7cebaba
15 lines
387 B
Python
15 lines
387 B
Python
from __future__ import absolute_import
|
|
from __future__ import division
|
|
from __future__ import print_function
|
|
from __future__ import unicode_literals
|
|
|
|
import operator_benchmark as op_bench
|
|
from pt import ( # noqa
|
|
unary_test, # noqa
|
|
)
|
|
import benchmark_all_other_test # noqa
|
|
import benchmark_all_quantized_test # noqa
|
|
|
|
if __name__ == "__main__":
|
|
op_bench.benchmark_runner.main()
|