Skip to content

Commit 09c0f8c

Browse files
committed
fix auto-merge
1 parent a7a021d commit 09c0f8c

File tree

2 files changed

+0
-4
lines changed

2 files changed

+0
-4
lines changed

test/dtypes/test_affine_quantized.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -96,7 +96,6 @@ def test_tensor_core_layout_transpose(self):
9696
aqt_shape = aqt.shape
9797
self.assertEqual(aqt_shape, shape)
9898

99-
@skip_if_rocm("ROCm development in progress")
10099
@unittest.skipIf(not torch.cuda.is_available(), "Need CUDA available")
101100
@common_utils.parametrize(
102101
"apply_quant",
@@ -181,7 +180,6 @@ def apply_uint6_weight_only_quant(linear):
181180
@common_utils.parametrize(
182181
"apply_quant", get_quantization_functions(is_cusparselt_available, True)
183182
)
184-
@skip_if_rocm("ROCm development in progress")
185183
@unittest.skipIf(not torch.cuda.is_available(), "Need CUDA available")
186184
@skip_if_rocm("ROCm enablement in progress")
187185
def test_print_quantized_module(self, apply_quant):
@@ -194,7 +192,6 @@ class TestAffineQuantizedBasic(TestCase):
194192
COMMON_DEVICES = ["cpu"] + (["cuda"] if torch.cuda.is_available() else [])
195193
COMMON_DTYPES = [torch.bfloat16]
196194

197-
@skip_if_rocm("ROCm development in progress")
198195
@common_utils.parametrize("device", COMMON_DEVICES)
199196
@common_utils.parametrize("dtype", COMMON_DTYPES)
200197
@skip_if_rocm("ROCm enablement in progress")

test/dtypes/test_floatx.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -108,7 +108,6 @@ def test_to_copy_device(self, ebits, mbits):
108108
@parametrize("ebits,mbits", _Floatx_DTYPES)
109109
@parametrize("bias", [False, True])
110110
@parametrize("dtype", [torch.half, torch.bfloat16])
111-
@skip_if_rocm("ROCm development in progress")
112111
@unittest.skipIf(is_fbcode(), reason="broken in fbcode")
113112
@skip_if_rocm("ROCm enablement in progress")
114113
def test_fpx_weight_only(self, ebits, mbits, bias, dtype):

0 commit comments

Comments
 (0)