mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00

Files

Xuehai Pan c0ed38e644 [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-17 14:34:42 +00:00

__init__.py

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

2024-07-17 14:34:42 +00:00

bench.py

Apply UFMT to all files in benchmarks/ (#105928 )

2023-07-26 01:18:48 +00:00

cells.py

Apply UFMT to all files in benchmarks/ (#105928 )

2023-07-26 01:18:48 +00:00

conftest.py

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

2024-07-17 14:34:42 +00:00

custom_lstms.py

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

2024-07-17 14:34:42 +00:00

factory.py

Apply UFMT to all files in benchmarks/ (#105928 )

2023-07-26 01:18:48 +00:00

fuser.py

Apply UFMT to all files in benchmarks/ (#105928 )

2023-07-26 01:18:48 +00:00

profile.py

Apply UFMT to all files in benchmarks/ (#105928 )

2023-07-26 01:18:48 +00:00

README.md

Lint trailing newlines (#54737 )

2021-03-30 13:09:52 -07:00

runner.py

[5/N][Easy] fix typo for usort config in pyproject.toml (kown -> known): sort torch (#127126 )

2024-05-27 14:49:57 +00:00

scratch.py

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

2024-07-17 14:34:42 +00:00

test_bench.py

[5/N][Easy] fix typo for usort config in pyproject.toml (kown -> known): sort torch (#127126 )

2024-05-27 14:49:57 +00:00

test.py

[Ez][BE]: Enable new stable ruff rules (#129825 )

2024-07-02 14:47:10 +00:00

README.md

Fast RNN benchmarks

Benchmarks for TorchScript models

For most stable results, do the following:

Set CPU Governor to performance mode (as opposed to energy save)
Turn off turbo for all CPUs (assuming Intel CPUs)
Shield cpus via cset shield when running benchmarks.

Some of these scripts accept command line args but most of them do not because I was lazy. They will probably be added sometime in the future, but the default sizes are pretty reasonable.

Test fastrnns (fwd + bwd) correctness

Test the fastrnns benchmarking scripts with the following: python -m fastrnns.test or run the test independently: python -m fastrnns.test --rnns jit

Run benchmarks

python -m fastrnns.bench

should give a good comparison, or you can specify the type of model to run

python -m fastrnns.bench --rnns cudnn aten jit --group rnns

Run model profiling, calls nvprof

python -m fastrnns.profile

should generate nvprof file for all models somewhere. you can also specify the models to generate nvprof files separately:

python -m fastrnns.profile --rnns aten jit

Caveats

Use Linux for the most accurate timing. A lot of these tests only run on CUDA.