diff --git a/paddle/fluid/operators/math/jit_kernel_test.cc b/paddle/fluid/operators/math/jit_kernel_test.cc index 667a95fe1a247cf9d3d63dae74f7e0fa9c2309ca..34fa2b9a7814dbd96de1e7c4a5be5a88978445bd 100644 --- a/paddle/fluid/operators/math/jit_kernel_test.cc +++ b/paddle/fluid/operators/math/jit_kernel_test.cc @@ -801,7 +801,11 @@ TEST(JitKernel, pool) { std::dynamic_pointer_cast(pvmul_d)); const auto& pvmul_from_key = jit::KernelPool::Instance().Get("vmulfjit4"); - EXPECT_EQ(pvmul_f, pvmul_from_key); +#if defined(__APPLE__) || defined(__OSX__) || defined(_WIN32) + EXPECT_EQ(pvmul_from_key, nullptr); +#else + EXPECT_EQ(pvmul_from_key, pvmul_f); +#endif const auto& pvmul_from_key2 = jit::KernelPool::Instance().Get("vmulfjit"); EXPECT_TRUE(pvmul_from_key2 == nullptr); } diff --git a/paddle/fluid/platform/init.cc b/paddle/fluid/platform/init.cc index 07abe1dd5c426e697d1598c9fa3e07cb48aa435a..2211e5504373b4a30e5fda0db22a41bdcd9f2421 100644 --- a/paddle/fluid/platform/init.cc +++ b/paddle/fluid/platform/init.cc @@ -116,6 +116,7 @@ void InitDevices(bool init_p2p, const std::vector devices) { platform::SetNumThreads(FLAGS_paddle_num_threads); #endif +#if !defined(_WIN32) && !defined(__APPLE__) && !defined(__OSX__) if (platform::jit::MayIUse(platform::jit::avx)) { #ifndef __AVX__ LOG(WARNING) << "AVX is available, Please re-compile on local machine"; @@ -157,8 +158,9 @@ void InitDevices(bool init_p2p, const std::vector devices) { AVX_GUIDE(AVX, NonAVX); } #endif - #undef AVX_GUIDE + +#endif } void InitGLOG(const std::string &prog_name) {