fix avx option (#16684)

test=release/1.4

fix avx option (#16684)
test=release/1.4
2b800923 · tensor-tang · GitHub · c6e6cdc9 · 2b800923
隐藏空白更改
内联并排

Showing with 9 addition and 7 deletion

paddle/fluid/operators/jit/test.cc paddle/fluid/operators/jit/test.cc +9 -7

未找到文件。
--- a/paddle/fluid/operators/jit/test.cc
+++ b/paddle/fluid/operators/jit/test.cc
@@ -991,15 +991,17 @@ TEST(JITKernel_pool, jitpool) {

 TEST(JITKernel_pool, more) {
  const auto& kers = jit::KernelPool::Instance().AllKernels();
-#if defined(__APPLE__) || defined(__OSX__)
-  EXPECT_EQ(kers.size(), 10UL);
-#else
-#ifdef PADDLE_WITH_MKLML
-  EXPECT_EQ(kers.size(), 22UL);
-#else
-  EXPECT_EQ(kers.size(), 8UL);
+  size_t target_num = 8;
+
+#ifdef __AVX__
+  target_num += 2;
 #endif
+
+#ifdef PADDLE_WITH_MKLML
+  target_num += 12;
 #endif
+
+  EXPECT_EQ(kers.size(), target_num);
 }

 TEST(JITKernel_pool, refer) {