Update test

Author: matthewdouglas

tests/test_parametrize.py

@@ -37,10 +37,7 @@ def __init__(self, device="cpu", dtype=torch.float32):
 @pytest.mark.parametrize("dtype", [torch.float32, torch.float16, torch.bfloat16], ids=describe_dtype)
 @pytest.mark.parametrize("quant_type", ["nf4", "fp4"])
 @pytest.mark.parametrize("compress_statistics", TRUE_FALSE, ids=id_formatter("compress_statistics"))
-@pytest.mark.parametrize(
-    "blocksize",
-    [64, 128, 256] if not ROCM_WARP_SIZE_64 else [128, 256],
-)
+@pytest.mark.parametrize("blocksize", [64, 128, 256])
 def test_replace_parameter_4bit(device, dtype, quant_type, compress_statistics, blocksize):
     """Test basic parameter replacement with 4-bit quantization on different dtypes."""
     if device == "hpu" and not is_supported_on_hpu(quant_type, dtype):