skip fbgemm config tests for 2.5 and below

jerryzh168 · jerryzh168 · commit d2066dc01744 · 2025-05-27T21:39:23.000-07:00
diff --git a/test/quantization/test_config_serialization.py b/test/quantization/test_config_serialization.py
@@ -35,6 +35,7 @@
     UIntXWeightOnlyConfig,
 )
 from torchao.sparsity.sparse_api import BlockSparseWeightConfig, SemiSparseWeightConfig
+from torchao.utils import TORCH_VERSION_AT_LEAST_2_6
 
 # Define test configurations as fixtures
 configs = [
@@ -78,9 +79,11 @@
             "linear2": Int8DynamicActivationInt4WeightConfig(),
         }
     ),
-    FbgemmConfig(torch.bfloat16, torch.int4, torch.bfloat16, [1, 1, 256]),
 ]
 
+if TORCH_VERSION_AT_LEAST_2_6:
+    configs += [FbgemmConfig(torch.bfloat16, torch.int4, torch.bfloat16, [1, 1, 256])]
+
 
 # Create ids for better test naming
 def get_config_ids(configs):

Original file line number	Diff line number	Diff line change
`@@ -35,6 +35,7 @@`
`35`	`35`	`UIntXWeightOnlyConfig,`
`36`	`36`	`)`
`37`	`37`	`from torchao.sparsity.sparse_api import BlockSparseWeightConfig, SemiSparseWeightConfig`
	`38`	`+from torchao.utils import TORCH_VERSION_AT_LEAST_2_6`
`38`	`39`
`39`	`40`	`# Define test configurations as fixtures`
`40`	`41`	`configs = [`
`@@ -78,9 +79,11 @@`
`78`	`79`	`"linear2": Int8DynamicActivationInt4WeightConfig(),`
`79`	`80`	`}`
`80`	`81`	`),`
`81`		`- FbgemmConfig(torch.bfloat16, torch.int4, torch.bfloat16, [1, 1, 256]),`
`82`	`82`	`]`
`83`	`83`
	`84`	`+if TORCH_VERSION_AT_LEAST_2_6:`
	`85`	`+ configs += [FbgemmConfig(torch.bfloat16, torch.int4, torch.bfloat16, [1, 1, 256])]`
	`86`	`+`
`84`	`87`
`85`	`88`	`# Create ids for better test naming`
`86`	`89`	`def get_config_ids(configs):`