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

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

Browse Source 
The `usort` config in `pyproject.toml` has no effect due to a typo. Fixing the typo make `usort` do more and generate the changes in the PR. Except `pyproject.toml`, all changes are generated by `lintrunner -a --take UFMT --all-files`.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/127126
Approved by: https://github.com/kit1980
ghstack dependencies: #127122, #127123, #127124, #127125
This commit is contained in:
Xuehai Pan
2024-05-25 16:21:09 +00:00
committed by PyTorch MergeBot
parent 4fdbaa794f
commit 7763c83af6
295 changed files with 572 additions and 467 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
benchmarks/transformer/sdpa.py
View File

@ -4,12 +4,13 @@ from contextlib import nullcontext
from dataclasses import asdict, dataclass
from typing import Callable, List, Tuple
from tabulate import tabulate
from tqdm import tqdm
import torch
import torch.utils.benchmark as benchmark
from tabulate import tabulate
from torch.nn.attention import sdpa_kernel, SDPBackend
from torch.nn.functional import scaled_dot_product_attention
from tqdm import tqdm
def benchmark_torch_function_in_microseconds(func: Callable, *args, **kwargs) -> float: