Remove fpx

Author: jainapurva

test/dtypes/test_affine_quantized_tensor_parallel.py

@@ -12,7 +12,6 @@
 from torchao.quantization import (
     float8_dynamic_activation_float8_weight,
     float8_weight_only,
-    fpx_weight_only,
     int4_weight_only,
     int8_dynamic_activation_int8_weight,
     int8_weight_only,
@@ -179,22 +178,10 @@ def test_tp(self, dtype):
         return self._test_tp(dtype)
 
 
-class TestFpxwoAffineQuantizedTensorParallel(TestAffineQuantizedTensorParallel):
-    QUANT_METHOD_FN = staticmethod(fpx_weight_only)
-    COMMON_DTYPES = [torch.bfloat16]
-
-    @common_utils.parametrize("dtype", COMMON_DTYPES)
-    @with_comms
-    @unittest.skipIf(not torch.cuda.is_available(), "Need CUDA available")
-    def test_tp(self, dtype):
-        return self._test_tp(dtype)
-
-
 common_utils.instantiate_parametrized_tests(TestInt8woAffineQuantizedTensorParallel)
 common_utils.instantiate_parametrized_tests(TestInt4woAffineQuantizedTensorParallel)
 common_utils.instantiate_parametrized_tests(TestGemliteLayoutTensorParallel)
 common_utils.instantiate_parametrized_tests(TestInt8dqAffineQuantizedTensorParallel)
-common_utils.instantiate_parametrized_tests(TestFpxwoAffineQuantizedTensorParallel)
 
 # Run only on H100
 if torch.cuda.is_available() and torch.cuda.get_device_capability() >= (9, 0):