Skip to content

Commit 4cbf9f3

Browse files
authored
[SmokeTest] Test antialias on MPS device (#8996)
It should be supported starting from 2.7, same way as on any other backend And with that, both CPU and MPS backends report the same confidence value: ``` German shepherd (cpu): 37.6% German shepherd (mps): 37.6% ```
1 parent 8ea4772 commit 4cbf9f3

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

test/smoke_test.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -96,7 +96,7 @@ def smoke_test_torchvision_resnet50_classify(device: str = "cpu") -> None:
9696
model.eval()
9797

9898
# Step 2: Initialize the inference transforms
99-
preprocess = weights.transforms(antialias=(device != "mps")) # antialias not supported on MPS
99+
preprocess = weights.transforms(antialias=True)
100100

101101
# Step 3: Apply inference preprocessing transforms
102102
batch = preprocess(img).unsqueeze(0)

0 commit comments

Comments
 (0)