Skip to content

Commit

Permalink
Pass all args to pytest.main to propagate user options like -k (#1640)
Browse files Browse the repository at this point in the history
Pass all args to pytest.main to propage user options like -k

Tested locally with `python test/test_ops.py -k test_dequantize_tensor_core_tiled_layout_correctness_quant_dequant`

which previously just ran all the tests but after this PR will run 60, the same number as `pytest test/test_ops.py -k test_dequantize_tensor_core_tiled_layout_correctness_quant_dequant`
  • Loading branch information
janeyx99 authored Jan 29, 2025
1 parent 7b0d2ce commit 2aed684
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion test/test_ops.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,5 @@
import itertools
import sys

import pytest
import torch
Expand Down Expand Up @@ -614,4 +615,4 @@ def test_marlin_qqq(batch_size, k_chunk, n_chunk, num_bits, group_size, mnk_fact


if __name__ == "__main__":
pytest.main([__file__])
pytest.main(sys.argv)

0 comments on commit 2aed684

Please sign in to comment.