Skip to content

Commit a36bd1d

Browse files
authored
Merge pull request #15 from ROCm/gemv_skip_comment
Update gemv skip comment
2 parents f4ac9ac + 485ba8f commit a36bd1d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/test_functional.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2375,7 +2375,7 @@ def test_normal_map_tree():
23752375
@pytest.mark.parametrize("kind", ['fc1', 'fc2', 'attn', 'attn_packed'], ids=['fc1', 'fc2', 'attn', 'attn_packed'])
23762376
@pytest.mark.parametrize("dtype", [torch.float16, torch.bfloat16, torch.float32], ids=['fp16', 'bf16', 'fp32'])
23772377
@pytest.mark.parametrize("quant_storage", [torch.uint8, torch.float16, torch.bfloat16, torch.float32], ids=['uint8', 'fp16', 'bf16', 'fp32'])
2378-
@pytest.mark.skipif(HIP_ENVIRONMENT, reason="this test is not supported on ROCm yet")
2378+
@pytest.mark.skipif(HIP_ENVIRONMENT, reason="gemv 4bit tests are partially enabled on MI300, others being fixed for warpsize 64")
23792379
def test_gemv_4bit(dtype, storage_type, quant_storage, double_quant, kind):
23802380
for dim in [128, 256, 512, 1024]:
23812381
#for dim in [4*1024]:

0 commit comments

Comments
 (0)