frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-11-11 22:34:53 +08:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source 
This reverts commit 7763c83af67eebfdd5185dbe6ce15ece2b992a0f.

Reverted https://github.com/pytorch/pytorch/pull/127126 on behalf of https://github.com/XuehaiPan due to Broken CI ([comment](https://github.com/pytorch/pytorch/pull/127126#issuecomment-2133044286))
This commit is contained in:
PyTorch MergeBot
2024-05-27 09:22:08 +00:00
parent 4608971f7a
commit 55c0ab2887
295 changed files with 468 additions and 573 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
benchmarks/transformer/attention_bias_benchmarks.py
View File

@ -4,15 +4,14 @@ from functools import partial
from typing import Callable, List, Union
import numpy as np
from tabulate import tabulate
from tqdm import tqdm
import torch
import torch.nn as nn
import torch.nn.functional as F
import torch.utils.benchmark as benchmark
from tabulate import tabulate
from torch.nn.attention.bias import CausalBias, CausalVariant
from torch.nn.parameter import Parameter
from tqdm import tqdm
def benchmark_torch_function_in_microseconds(func: Callable, *args, **kwargs) -> float: