From 09d120dc734c8d5d572620efebd8a9375358b814 Mon Sep 17 00:00:00 2001 From: alanhuangyoo Date: Mon, 24 Aug 2026 19:43:58 +0800 Subject: [PATCH] Gate MX/NVFP4 safetensors test configs on sm100 The MX and NVFP4 entries in _ALL_TEST_CONFIGS are added based on the torch version alone, but both kernels require sm100+. On sm89/sm90 the parametrized cases run and fail instead of being skipped. --- test/prototype/safetensors/test_safetensors_support.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/test/prototype/safetensors/test_safetensors_support.py b/test/prototype/safetensors/test_safetensors_support.py index e40f8cb862..5d0205dcc7 100644 --- a/test/prototype/safetensors/test_safetensors_support.py +++ b/test/prototype/safetensors/test_safetensors_support.py @@ -59,8 +59,8 @@ (Int4WeightOnlyConfig(int4_packing_format="tile_packed_to_4d"), False), ] - # MX and NVFP4 configs require torch >= 2.11 - if torch_version_at_least("2.11.0.dev"): + # MX and NVFP4 configs require torch >= 2.11 and sm100+ + if torch_version_at_least("2.11.0.dev") and is_sm_at_least_100(): _ALL_TEST_CONFIGS += [ (MXDynamicActivationMXWeightConfig(), False), (NVFP4DynamicActivationNVFP4WeightConfig(), False),