Skip to content

Commit 942fe2e

Browse files
committed
skip pined memory for xpu
1 parent e004ced commit 942fe2e

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

test/quantization/quantize_/workflows/int8/test_int8_tensor.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -225,6 +225,7 @@ def test_available_gpu_kernels(self):
225225
"extern_kernels._int_mm", 1
226226
).check_count("triton_poi_fused", 1).run(code[0])
227227

228+
@unittest.skipIf(not torch.cuda.is_available(), "Need CUDA available")
228229
@common_utils.parametrize("config", INT8_TEST_CONFIGS)
229230
def test_pin_memory(self, config):
230231
linear = torch.nn.Linear(

0 commit comments

Comments
 (0)