We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 568e2ad commit bb3f03bCopy full SHA for bb3f03b
test/prototype/test_smoothquant.py
@@ -61,7 +61,12 @@ def forward(self, x):
61
return x
62
63
64
-device_list = ["cpu", "cuda"] if torch.cuda.is_available() else ["cpu"]
+device_list = ["cpu"]
65
+if torch.cuda.is_available():
66
+ device_list.append("cuda")
67
+
68
+if torch.xpu.is_available():
69
+ device_list.append("xpu")
70
71
72
@unittest.skipIf(torch.version.hip is not None, "Skipping tests in ROCm")
0 commit comments