File size: 1,256 Bytes
20347e1 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 | [
"tests/test_bnb_mps.py::test_dequantize_matches_reference[128]",
"tests/test_bnb_mps.py::test_dequantize_matches_reference[64]",
"tests/test_bnb_mps.py::test_gemm_correctness[1-128]",
"tests/test_bnb_mps.py::test_gemm_correctness[1-64]",
"tests/test_bnb_mps.py::test_gemm_correctness[2-128]",
"tests/test_bnb_mps.py::test_gemm_correctness[2-64]",
"tests/test_bnb_mps.py::test_gemv_correctness[1-128]",
"tests/test_bnb_mps.py::test_gemv_correctness[1-64]",
"tests/test_bnb_mps.py::test_gemv_correctness[2-128]",
"tests/test_bnb_mps.py::test_gemv_correctness[2-64]",
"tests/test_bnb_mps.py::test_linear_4bit_auto_select",
"tests/test_bnb_mps.py::test_quantize_dequantize_roundtrip[dtype0-1-128]",
"tests/test_bnb_mps.py::test_quantize_dequantize_roundtrip[dtype0-1-64]",
"tests/test_bnb_mps.py::test_quantize_dequantize_roundtrip[dtype0-2-128]",
"tests/test_bnb_mps.py::test_quantize_dequantize_roundtrip[dtype0-2-64]",
"tests/test_bnb_mps.py::test_quantize_dequantize_roundtrip[dtype1-1-128]",
"tests/test_bnb_mps.py::test_quantize_dequantize_roundtrip[dtype1-1-64]",
"tests/test_bnb_mps.py::test_quantize_dequantize_roundtrip[dtype1-2-128]",
"tests/test_bnb_mps.py::test_quantize_dequantize_roundtrip[dtype1-2-64]"
] |