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

[BE] Remove macos-13 guard from bench_mps_ops (#159732)

Browse Source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/159732
Approved by: https://github.com/dcci
ghstack dependencies: #159731
This commit is contained in:
Nikita Shulga
2025-08-03 12:23:21 -07:00
committed by PyTorch MergeBot
parent fecdebe385
commit 5116c49b52
1 changed files with 1 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
test/bench_mps_ops.py
View File

@ -154,9 +154,7 @@ def bench_scan(
def main() -> None:
dtypes = [torch.float16, torch.float32]
if torch.backends.mps.is_macos_or_newer(14, 0):
dtypes.append(torch.bfloat16)
dtypes = [torch.float16, torch.float32, torch.bfloat16]
# Profile index ops
B = 11