saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-04 04:07:22 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Disable QOrderedMatMul with bias tests on Windows (#12901)

Browse source
This commit is contained in:
Hariharan Seshadri 2022-09-08 17:57:37 -07:00 committed by GitHub
parent b5327595f3
commit 0b235b2763
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 12 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
onnxruntime/test/contrib_ops/qordered_matmul_op_test.cc
View file

@ -122,6 +122,7 @@ TEST(QOrderedTest, MatMul_COL_16x64x32) {
false /* add bias */, false /* broadcast batch c */);
}
#ifndef _WIN32
TEST(QOrderedTest, MatMul_bias_COL_16x64x32) {
std::vector<int64_t> shape_A = {16, 32};
std::vector<int64_t> shape_B = {32, 64};
@ -130,6 +131,7 @@ TEST(QOrderedTest, MatMul_bias_COL_16x64x32) {
1.0f / 32.0f, 1.0f / 32.0f, 0.0f /*scaleC*/, 2.0f,
true /* add bias */, false /* broadcast batch c */);
}
#endif // ifndef _WIN32
TEST(QOrderedTest, MatMul_addC_COL_16x64x32) {
std::vector<int64_t> shape_A = {16, 32};
@ -140,6 +142,7 @@ TEST(QOrderedTest, MatMul_addC_COL_16x64x32) {
false /* add bias */, false /* broadcast batch c */);
}
#ifndef _WIN32
TEST(QOrderedTest, MatMul_bias_addC_COL_16x64x32) {
std::vector<int64_t> shape_A = {16, 32};
std::vector<int64_t> shape_B = {32, 64};
@ -148,7 +151,8 @@ TEST(QOrderedTest, MatMul_bias_addC_COL_16x64x32) {
1.0f / 32.0f, 1.0f / 32.0f, 4.0f /*scaleC*/, 2.0f,
true /* add bias */, true /* broadcast batch c */);
}
#endif // ifndef _WIN32
TEST(QOrderedTest, MatMul_COL_16x64x32_b3_1) {
std::vector<int64_t> shape_A = {3, 16, 32};
std::vector<int64_t> shape_B = {1, 32, 64};
@ -158,6 +162,7 @@ TEST(QOrderedTest, MatMul_COL_16x64x32_b3_1) {
false /* add bias */, false /* broadcast batch c */);
}
#ifndef _WIN32
TEST(QOrderedTest, MatMul_bias_COL_16x64x32_b2_1) {
std::vector<int64_t> shape_A = {2, 16, 32};
std::vector<int64_t> shape_B = {1, 32, 64};
@ -166,6 +171,7 @@ TEST(QOrderedTest, MatMul_bias_COL_16x64x32_b2_1) {
1.0f / 32.0f, 1.0f / 32.0f, 0.0f /*scaleC*/, 2.0f,
true /* add bias */, false /* broadcast batch c */);
}
#endif // ifndef _WIN32
TEST(QOrderedTest, MatMul_addC_COL_16x64x32_b2_1) {
std::vector<int64_t> shape_A = {2, 16, 32};
@ -185,6 +191,7 @@ TEST(QOrderedTest, MatMul_addC_broadcastC_COL_16x64x32_b2_1) {
false /* add bias */, true /* broadcast batch c */);
}
#ifndef _WIN32
TEST(QOrderedTest, MatMul_addC_bias_COL_16x64x32_b2_1) {
std::vector<int64_t> shape_A = {2, 16, 32};
std::vector<int64_t> shape_B = {1, 32, 64};
@ -193,7 +200,9 @@ TEST(QOrderedTest, MatMul_addC_bias_COL_16x64x32_b2_1) {
1.0f / 32.0f, 1.0f / 32.0f, 0.0f /*scaleC*/, 2.0f,
true /* add bias */, false /* broadcast batch c */);
}
#endif // ifndef _WIN32
#ifndef _WIN32
TEST(QOrderedTest, MatMul_bias_addC_broadcastC_COL_16x64x32_b2_1) {
std::vector<int64_t> shape_A = {2, 16, 32};
std::vector<int64_t> shape_B = {1, 32, 64};
@ -202,8 +211,9 @@ TEST(QOrderedTest, MatMul_bias_addC_broadcastC_COL_16x64x32_b2_1) {
1.0f / 32.0f, 1.0f / 32.0f, 0.0f /*scaleC*/, 2.0f,
true /* add bias */, true /* broadcast batch c */);
}
#endif // ifndef _WIN32
#endif
#endif // USE_CUDA
} // namespace test
} // namespace onnxruntime