Skip to content

Commit

Permalink
testing ci
Browse files Browse the repository at this point in the history
  • Loading branch information
@fajin-corp
fajin-corp committed Nov 8, 2024
1 parent adce284 commit 57ef96b
Showing 1 changed file with 75 additions and 75 deletions.
150 changes: 75 additions & 75 deletions onnxruntime/test/contrib_ops/matmul_4bits_test.cc
View file
Original file line number Diff line number Diff line change
Expand Up @@ -394,81 +394,81 @@ TEST(MatMulNBits, Float32_Accuracy4) {
#if !defined(USE_DML) && (defined(MLAS_TARGET_AMD64_IX86) || defined(MLAS_TARGET_ARM64))
// Actual and expected difference is over 0.01 with DmlExecutionProvider.
// Skip the tests instead of raising the tolerance to make is pass.
TEST(MatMulNBits, Float16_Accuracy0) {
TestMatMulNBitsTyped<MLFloat16, 1, 1, 16, 16, 0>();
TestMatMulNBitsTyped<MLFloat16, 1, 288, 93, 32, 0>();
TestMatMulNBitsTyped<MLFloat16, 1, 288, 1234, 16, 0>();
TestMatMulNBitsTyped<MLFloat16, 2, 1, 16, 16, 0>();
TestMatMulNBitsTyped<MLFloat16, 100, 2, 16, 16, 0>();
TestMatMulNBitsTyped<MLFloat16, 100, 288, 1024, 128, 0>();
TestMatMulNBitsTyped<MLFloat16, 100, 288, 93, 32, 0>();
TestMatMulNBitsTyped<MLFloat16, 100, 288, 1234, 16, 0>();
}

TEST(MatMulNBits, Float16_Accuracy1) {
TestMatMulNBitsTyped<MLFloat16, 1, 1, 16, 16, 1>();
TestMatMulNBitsTyped<MLFloat16, 1, 288, 93, 32, 1>();
TestMatMulNBitsTyped<MLFloat16, 1, 288, 1234, 16, 1>();
TestMatMulNBitsTyped<MLFloat16, 2, 1, 16, 16, 1>();
TestMatMulNBitsTyped<MLFloat16, 100, 2, 16, 16, 1>();
TestMatMulNBitsTyped<MLFloat16, 100, 288, 1024, 128, 1>();
TestMatMulNBitsTyped<MLFloat16, 100, 288, 93, 32, 1>();
TestMatMulNBitsTyped<MLFloat16, 100, 288, 1234, 16, 1>();
}

TEST(MatMulNBits, Float16_Accuracy2) {
TestMatMulNBitsTyped<MLFloat16, 1, 1, 16, 16, 2>();
TestMatMulNBitsTyped<MLFloat16, 1, 2, 16, 16, 2>();
TestMatMulNBitsTyped<MLFloat16, 1, 32, 16, 16, 2>();
TestMatMulNBitsTyped<MLFloat16, 1, 32, 32, 16, 2>();
TestMatMulNBitsTyped<MLFloat16, 1, 32, 16, 128, 2>();
TestMatMulNBitsTyped<MLFloat16, 1, 288, 16, 16, 2>();
TestMatMulNBitsTyped<MLFloat16, 1, 288, 1024, 16, 2>();
TestMatMulNBitsTyped<MLFloat16, 1, 288, 1024, 128, 2>();
TestMatMulNBitsTyped<MLFloat16, 1, 288, 93, 32, 2>();
TestMatMulNBitsTyped<MLFloat16, 1, 288, 93, 128, 2>();
TestMatMulNBitsTyped<MLFloat16, 1, 288, 1234, 16, 2>();
TestMatMulNBitsTyped<MLFloat16, 2, 1, 16, 16, 2>();
TestMatMulNBitsTyped<MLFloat16, 2, 2, 16, 16, 2>();
TestMatMulNBitsTyped<MLFloat16, 100, 1, 16, 16, 2>();
TestMatMulNBitsTyped<MLFloat16, 100, 2, 16, 16, 2>();
TestMatMulNBitsTyped<MLFloat16, 100, 32, 16, 16, 2>();
TestMatMulNBitsTyped<MLFloat16, 100, 32, 32, 16, 2>();
TestMatMulNBitsTyped<MLFloat16, 100, 32, 16, 128, 2>();
TestMatMulNBitsTyped<MLFloat16, 100, 288, 16, 16, 2>();
TestMatMulNBitsTyped<MLFloat16, 100, 288, 1024, 16, 2>();
TestMatMulNBitsTyped<MLFloat16, 100, 288, 1024, 128, 2>();
TestMatMulNBitsTyped<MLFloat16, 100, 288, 93, 32, 2>();
TestMatMulNBitsTyped<MLFloat16, 100, 288, 93, 128, 2>();
TestMatMulNBitsTyped<MLFloat16, 100, 288, 1234, 16, 2>();
}

TEST(MatMulNBits, Float16_Accuracy4) {
TestMatMulNBitsTyped<MLFloat16, 1, 1, 16, 16, 4>();
TestMatMulNBitsTyped<MLFloat16, 1, 2, 16, 16, 4>();
TestMatMulNBitsTyped<MLFloat16, 1, 32, 16, 16, 4>();
TestMatMulNBitsTyped<MLFloat16, 1, 32, 32, 16, 4>();
TestMatMulNBitsTyped<MLFloat16, 1, 32, 16, 128, 4>();
TestMatMulNBitsTyped<MLFloat16, 1, 288, 16, 16, 4>();
TestMatMulNBitsTyped<MLFloat16, 1, 288, 1024, 16, 4>();
TestMatMulNBitsTyped<MLFloat16, 1, 288, 1024, 128, 4>();
TestMatMulNBitsTyped<MLFloat16, 1, 288, 93, 32, 4>();
TestMatMulNBitsTyped<MLFloat16, 1, 288, 93, 128, 4>();
TestMatMulNBitsTyped<MLFloat16, 1, 288, 1234, 16, 4>();
TestMatMulNBitsTyped<MLFloat16, 2, 1, 16, 16, 4>();
TestMatMulNBitsTyped<MLFloat16, 2, 2, 16, 16, 4>();
TestMatMulNBitsTyped<MLFloat16, 100, 1, 16, 16, 4>();
TestMatMulNBitsTyped<MLFloat16, 100, 2, 16, 16, 4>();
TestMatMulNBitsTyped<MLFloat16, 100, 32, 16, 16, 4>();
TestMatMulNBitsTyped<MLFloat16, 100, 32, 32, 16, 4>();
TestMatMulNBitsTyped<MLFloat16, 100, 32, 16, 128, 4>();
TestMatMulNBitsTyped<MLFloat16, 100, 288, 16, 16, 4>();
TestMatMulNBitsTyped<MLFloat16, 100, 288, 1024, 16, 4>();
TestMatMulNBitsTyped<MLFloat16, 100, 288, 1024, 128, 4>();
TestMatMulNBitsTyped<MLFloat16, 100, 288, 93, 32, 4>();
TestMatMulNBitsTyped<MLFloat16, 100, 288, 93, 128, 4>();
TestMatMulNBitsTyped<MLFloat16, 100, 288, 1234, 16, 4>();
}
// TEST(MatMulNBits, Float16_Accuracy0) {
// TestMatMulNBitsTyped<MLFloat16, 1, 1, 16, 16, 0>();
// TestMatMulNBitsTyped<MLFloat16, 1, 288, 93, 32, 0>();
// TestMatMulNBitsTyped<MLFloat16, 1, 288, 1234, 16, 0>();
// TestMatMulNBitsTyped<MLFloat16, 2, 1, 16, 16, 0>();
// TestMatMulNBitsTyped<MLFloat16, 100, 2, 16, 16, 0>();
// TestMatMulNBitsTyped<MLFloat16, 100, 288, 1024, 128, 0>();
// TestMatMulNBitsTyped<MLFloat16, 100, 288, 93, 32, 0>();
// TestMatMulNBitsTyped<MLFloat16, 100, 288, 1234, 16, 0>();
// }

// TEST(MatMulNBits, Float16_Accuracy1) {
// TestMatMulNBitsTyped<MLFloat16, 1, 1, 16, 16, 1>();
// TestMatMulNBitsTyped<MLFloat16, 1, 288, 93, 32, 1>();
// TestMatMulNBitsTyped<MLFloat16, 1, 288, 1234, 16, 1>();
// TestMatMulNBitsTyped<MLFloat16, 2, 1, 16, 16, 1>();
// TestMatMulNBitsTyped<MLFloat16, 100, 2, 16, 16, 1>();
// TestMatMulNBitsTyped<MLFloat16, 100, 288, 1024, 128, 1>();
// TestMatMulNBitsTyped<MLFloat16, 100, 288, 93, 32, 1>();
// TestMatMulNBitsTyped<MLFloat16, 100, 288, 1234, 16, 1>();
// }

// TEST(MatMulNBits, Float16_Accuracy2) {
// TestMatMulNBitsTyped<MLFloat16, 1, 1, 16, 16, 2>();
// TestMatMulNBitsTyped<MLFloat16, 1, 2, 16, 16, 2>();
// TestMatMulNBitsTyped<MLFloat16, 1, 32, 16, 16, 2>();
// TestMatMulNBitsTyped<MLFloat16, 1, 32, 32, 16, 2>();
// TestMatMulNBitsTyped<MLFloat16, 1, 32, 16, 128, 2>();
// TestMatMulNBitsTyped<MLFloat16, 1, 288, 16, 16, 2>();
// TestMatMulNBitsTyped<MLFloat16, 1, 288, 1024, 16, 2>();
// TestMatMulNBitsTyped<MLFloat16, 1, 288, 1024, 128, 2>();
// TestMatMulNBitsTyped<MLFloat16, 1, 288, 93, 32, 2>();
// TestMatMulNBitsTyped<MLFloat16, 1, 288, 93, 128, 2>();
// TestMatMulNBitsTyped<MLFloat16, 1, 288, 1234, 16, 2>();
// TestMatMulNBitsTyped<MLFloat16, 2, 1, 16, 16, 2>();
// TestMatMulNBitsTyped<MLFloat16, 2, 2, 16, 16, 2>();
// TestMatMulNBitsTyped<MLFloat16, 100, 1, 16, 16, 2>();
// TestMatMulNBitsTyped<MLFloat16, 100, 2, 16, 16, 2>();
// TestMatMulNBitsTyped<MLFloat16, 100, 32, 16, 16, 2>();
// TestMatMulNBitsTyped<MLFloat16, 100, 32, 32, 16, 2>();
// TestMatMulNBitsTyped<MLFloat16, 100, 32, 16, 128, 2>();
// TestMatMulNBitsTyped<MLFloat16, 100, 288, 16, 16, 2>();
// TestMatMulNBitsTyped<MLFloat16, 100, 288, 1024, 16, 2>();
// TestMatMulNBitsTyped<MLFloat16, 100, 288, 1024, 128, 2>();
// TestMatMulNBitsTyped<MLFloat16, 100, 288, 93, 32, 2>();
// TestMatMulNBitsTyped<MLFloat16, 100, 288, 93, 128, 2>();
// TestMatMulNBitsTyped<MLFloat16, 100, 288, 1234, 16, 2>();
// }

// TEST(MatMulNBits, Float16_Accuracy4) {
// TestMatMulNBitsTyped<MLFloat16, 1, 1, 16, 16, 4>();
// TestMatMulNBitsTyped<MLFloat16, 1, 2, 16, 16, 4>();
// TestMatMulNBitsTyped<MLFloat16, 1, 32, 16, 16, 4>();
// TestMatMulNBitsTyped<MLFloat16, 1, 32, 32, 16, 4>();
// TestMatMulNBitsTyped<MLFloat16, 1, 32, 16, 128, 4>();
// TestMatMulNBitsTyped<MLFloat16, 1, 288, 16, 16, 4>();
// TestMatMulNBitsTyped<MLFloat16, 1, 288, 1024, 16, 4>();
// TestMatMulNBitsTyped<MLFloat16, 1, 288, 1024, 128, 4>();
// TestMatMulNBitsTyped<MLFloat16, 1, 288, 93, 32, 4>();
// TestMatMulNBitsTyped<MLFloat16, 1, 288, 93, 128, 4>();
// TestMatMulNBitsTyped<MLFloat16, 1, 288, 1234, 16, 4>();
// TestMatMulNBitsTyped<MLFloat16, 2, 1, 16, 16, 4>();
// TestMatMulNBitsTyped<MLFloat16, 2, 2, 16, 16, 4>();
// TestMatMulNBitsTyped<MLFloat16, 100, 1, 16, 16, 4>();
// TestMatMulNBitsTyped<MLFloat16, 100, 2, 16, 16, 4>();
// TestMatMulNBitsTyped<MLFloat16, 100, 32, 16, 16, 4>();
// TestMatMulNBitsTyped<MLFloat16, 100, 32, 32, 16, 4>();
// TestMatMulNBitsTyped<MLFloat16, 100, 32, 16, 128, 4>();
// TestMatMulNBitsTyped<MLFloat16, 100, 288, 16, 16, 4>();
// TestMatMulNBitsTyped<MLFloat16, 100, 288, 1024, 16, 4>();
// TestMatMulNBitsTyped<MLFloat16, 100, 288, 1024, 128, 4>();
// TestMatMulNBitsTyped<MLFloat16, 100, 288, 93, 32, 4>();
// TestMatMulNBitsTyped<MLFloat16, 100, 288, 93, 128, 4>();
// TestMatMulNBitsTyped<MLFloat16, 100, 288, 1234, 16, 4>();
// }
#endif // !defined(USE_DML) && (defined(MLAS_TARGET_AMD64_IX86) || defined(MLAS_TARGET_ARM64))

#if defined(USE_CUDA) || defined(USE_ROCM) || defined(USE_DML) || defined(USE_WEBGPU)
Expand Down

0 comments on commit 57ef96b

Please sign in to comment.