Skip to content

Commit bb3f03b

Browse files
Enable SmoothQuant Test on Intel GPU (#3185)
Enable SmoothQuant Test on Intel GPU
1 parent 568e2ad commit bb3f03b

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

test/prototype/test_smoothquant.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -61,7 +61,12 @@ def forward(self, x):
6161
return x
6262

6363

64-
device_list = ["cpu", "cuda"] if torch.cuda.is_available() else ["cpu"]
64+
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")
6570

6671

6772
@unittest.skipIf(torch.version.hip is not None, "Skipping tests in ROCm")

0 commit comments

Comments
 (0)