Skip to content

Commit 5116c49

Browse files
malfetpytorchmergebot
authored andcommitted
[BE] Remove macos-13 guard from bench_mps_ops (pytorch#159732)
Pull Request resolved: pytorch#159732 Approved by: https://github.com/dcci ghstack dependencies: pytorch#159731
1 parent fecdebe commit 5116c49

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

test/bench_mps_ops.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -154,9 +154,7 @@ def f_1d(t):
154154

155155

156156
def main() -> None:
157-
dtypes = [torch.float16, torch.float32]
158-
if torch.backends.mps.is_macos_or_newer(14, 0):
159-
dtypes.append(torch.bfloat16)
157+
dtypes = [torch.float16, torch.float32, torch.bfloat16]
160158

161159
# Profile index ops
162160
B = 11

0 commit comments

Comments
 (0)