Skip to content

Commit 3fde9ea

Browse files
committed
skip WIP config for XPU
1 parent 77f23d0 commit 3fde9ea

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

test/prototype/safetensors/test_safetensors_support.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -49,7 +49,7 @@
4949

5050
# Int4WeightOnlyConfig with tinygemm uses sm90a CUTLASS kernels which are not
5151
# forward-compatible with sm100+
52-
if not is_sm_at_least_100():
52+
if not torch.xpu.is_available() and not is_sm_at_least_100():
5353
_ALL_TEST_CONFIGS += [
5454
(Int4WeightOnlyConfig(), False),
5555
(Int4WeightOnlyConfig(), True),
@@ -63,7 +63,7 @@
6363
]
6464

6565
# MX and NVFP4 configs require torch >= 2.11
66-
if torch_version_at_least("2.11.0.dev"):
66+
if not torch.xpu.is_available() and torch_version_at_least("2.11.0.dev"):
6767
_ALL_TEST_CONFIGS += [
6868
(MXDynamicActivationMXWeightConfig(), False),
6969
(NVFP4DynamicActivationNVFP4WeightConfig(), False),

0 commit comments

Comments
 (0)