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

[BE][Easy][3/19] enforce style for empty lines in import segments in benchmarks/ (#129754)

Browse Source 
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
This commit is contained in:
Xuehai Pan
2024-07-16 11:39:35 +08:00
committed by PyTorch MergeBot
parent 32995dec28
commit c0ed38e644
126 changed files with 163 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
benchmarks/framework_overhead_benchmark/framework_overhead_benchmark.py
View File

@ -1,10 +1,11 @@
import argparse
from pt_wrapper_module import WrapperModule
from SimpleAddModule import add_tensors_loop, SimpleAddModule
from utils import benchmark_module, BenchmarkConfig, ModuleConfig, ms_to_us
""" Framework overhead benchmark script.
Benchmark framework overhead.
Currently supported ops: add.

1
benchmarks/framework_overhead_benchmark/utils.py
View File

@ -3,6 +3,7 @@ from collections import namedtuple
from torch.utils import ThroughputBenchmark
NUM_LOOP_ITERS = 1000
BenchmarkConfig = namedtuple("BenchmarkConfig", "num_warmup_iters num_iters")
ModuleConfig = namedtuple("ModuleConfig", "pt_fn c2_op num_params graph_mode")