Skip to content

Commit

Permalink
Apply ruff FMT
Browse files Browse the repository at this point in the history
  • Loading branch information
ahadnagy committed Jan 7, 2025
1 parent ecd85fc commit 4493482
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -131,7 +131,7 @@ def test_marlin_int4_weight_qbits_tensor_linear(batch_size, tokens, in_features,
)


#Tests previous Marlin kernel bug: https://github.com/huggingface/optimum-quanto/issues/332
# Tests previous Marlin kernel bug: https://github.com/huggingface/optimum-quanto/issues/332
@pytest.mark.skipif(
not is_extension_available("quanto_cuda") or torch.cuda.get_device_capability()[0] < 8,
reason="CUDA >= sm80 not available",
Expand Down

0 comments on commit 4493482

Please sign in to comment.