Skip to content

Commit 2aed684

Browse files
authored
Pass all args to pytest.main to propagate user options like -k (#1640)
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`
1 parent 7b0d2ce commit 2aed684

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

test/test_ops.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
import itertools
2+
import sys
23

34
import pytest
45
import torch
@@ -614,4 +615,4 @@ def test_marlin_qqq(batch_size, k_chunk, n_chunk, num_bits, group_size, mnk_fact
614615

615616

616617
if __name__ == "__main__":
617-
pytest.main([__file__])
618+
pytest.main(sys.argv)

0 commit comments

Comments
 (0)