pytorch/benchmarks/operator_benchmark/operator_benchmark.py

# TODO (mingzhe09088): get rid of noqa
import benchmark_runner  # noqa: F401
from benchmark_pytorch import TorchBenchmarkBase  # noqa: F401
from benchmark_test_generator import *  # noqa: F401,F403

from benchmark_utils import *  # noqa: F401,F403
introduce a new interface to add op [core changes] (#21147) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/21147 This diff introduces a new interface to add PT/C2 operators to the benchmark suite. The following steps are needed to add a new operator: 1. Specify the input shapes, args to an operator in configs 2. Create a PT/C2 benchmark class which includes ```init``` (create tensors), ```forward``` (specify the operator to be tested.), and ```backward```(gradient of an op.) methods 3. call generate_pt_test/generate_c2_test to create test cases based on configs Reviewed By: zheng-xq Differential Revision: D15250380 fbshipit-source-id: 1025a7cf60d2427baa0f3f716455946d3d3e6a27 2019-05-31 16:08:09 +00:00			`# TODO (mingzhe09088): get rid of noqa`
			`import benchmark_runner # noqa: F401`
			`from benchmark_pytorch import TorchBenchmarkBase # noqa: F401`
			`from benchmark_test_generator import * # noqa: F401,F403`
[BE][Easy][3/19] enforce style for empty lines in import segments in `benchmarks/` (#129754) See https://github.com/pytorch/pytorch/pull/129751#issue-2380881501. Most changes are auto-generated by linter. You can review these PRs via: ```bash git diff --ignore-all-space --ignore-blank-lines HEAD~1 ``` Pull Request resolved: https://github.com/pytorch/pytorch/pull/129754 Approved by: https://github.com/ezyang 2024-07-16 03:39:35 +00:00
introduce a new interface to add op [core changes] (#21147) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/21147 This diff introduces a new interface to add PT/C2 operators to the benchmark suite. The following steps are needed to add a new operator: 1. Specify the input shapes, args to an operator in configs 2. Create a PT/C2 benchmark class which includes ```init``` (create tensors), ```forward``` (specify the operator to be tested.), and ```backward```(gradient of an op.) methods 3. call generate_pt_test/generate_c2_test to create test cases based on configs Reviewed By: zheng-xq Differential Revision: D15250380 fbshipit-source-id: 1025a7cf60d2427baa0f3f716455946d3d3e6a27 2019-05-31 16:08:09 +00:00			`from benchmark_utils import * # noqa: F401,F403`