saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-05-14 20:48:00 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

disable 2bit test

Browse source 
Signed-off-by: Liqun Fu <liqun.fu@microsoft.com>
This commit is contained in:
Liqun Fu 2025-02-03 14:05:51 -08:00
parent 0130061001
commit b4aad0134c
2 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
onnxruntime/test/contrib_ops/matmul_4bits_test.cc
View file

@ -394,6 +394,7 @@ TEST(MatMulNBits, Float32_Accuracy4) {
TestMatMulNBitsTyped<float, Q4Bits, 100, 288, 1234, 16, 4>();
}
// TODO: enable and add more tests for 2bit development.
TEST(MatMulNBits, DISABLED_Float32_Accuracy4_Q2) {
TestMatMulNBitsTyped<float, Q2Bits, 2, 1, 1, 32, 4>();
}

3
onnxruntime/test/mlas/unittest/test_sqnbitgemm.cpp
View file

@ -443,8 +443,9 @@ class SQNBitGemmShortExecuteTest : public MlasTestFixture<MlasSQNBitGemmTest<Blk
static size_t SQNBitGemmRegisterAllShortExecuteTests() {
size_t count = 0;
// TODO: enable these test for 2bit development.
//count += SQNBitGemmShortExecuteTest<2, 16>::RegisterShortExecuteTests();
count += SQNBitGemmShortExecuteTest<2, 32>::RegisterShortExecuteTests();
//count += SQNBitGemmShortExecuteTest<2, 32>::RegisterShortExecuteTests();
//count += SQNBitGemmShortExecuteTest<2, 64>::RegisterShortExecuteTests();
//count += SQNBitGemmShortExecuteTest<2, 128>::RegisterShortExecuteTests();
//count += SQNBitGemmShortExecuteTest<2, 256>::RegisterShortExecuteTests();