We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 77f23d0 commit 3fde9eaCopy full SHA for 3fde9ea
test/prototype/safetensors/test_safetensors_support.py
@@ -49,7 +49,7 @@
49
50
# Int4WeightOnlyConfig with tinygemm uses sm90a CUTLASS kernels which are not
51
# forward-compatible with sm100+
52
-if not is_sm_at_least_100():
+if not torch.xpu.is_available() and not is_sm_at_least_100():
53
_ALL_TEST_CONFIGS += [
54
(Int4WeightOnlyConfig(), False),
55
(Int4WeightOnlyConfig(), True),
@@ -63,7 +63,7 @@
63
]
64
65
# MX and NVFP4 configs require torch >= 2.11
66
-if torch_version_at_least("2.11.0.dev"):
+if not torch.xpu.is_available() and torch_version_at_least("2.11.0.dev"):
67
68
(MXDynamicActivationMXWeightConfig(), False),
69
(NVFP4DynamicActivationNVFP4WeightConfig(), False),
0 commit comments