update

PaddlePaddle · Apr 8, 2024 · defa974 · defa974
1 parent 156852f
commit defa974
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/paddle/fluid/pybind/pybind.cc b/paddle/fluid/pybind/pybind.cc
@@ -405,6 +405,10 @@ bool SupportsInt8() {
 #endif
 }
 
+bool SupportsAvx512F() {
+  return phi::backends::cpu::MayIUse(phi::backends::cpu::cpu_isa_t::avx512f);
+}
+
 bool SupportsVNNI() {
 #ifndef PADDLE_WITH_DNNL
   return false;
@@ -2153,6 +2157,7 @@ All parameter, weight, gradient are variables in Paddle.
   m.def("supports_bfloat16", SupportsBfloat16);
   m.def("supports_bfloat16_fast_performance", SupportsBfloat16FastPerformance);
   m.def("supports_int8", SupportsInt8);
+  m.def("supports_avx512f", SupportsAvx512F);
   m.def("supports_vnni", SupportsVNNI);
   m.def("op_supported_infos", imperative::OpSupportedInfos);
   m.def("is_compiled_with_brpc", IsCompiledWithBrpc);

diff --git a/test/legacy_test/test_fused_layernorm_op.py b/test/legacy_test/test_fused_layernorm_op.py
@@ -793,7 +793,7 @@ def test_residual_bias_add_layernorm_int8(self):
 
 
 @unittest.skipIf(
-    not core.avx_supported() or not core.is_compiled_with_avx(),
+    not core.supports_avx512f() or not core.is_compiled_with_avx(),
     "machine is not support AVX or is not compiled with AVX",
 )
 class TestlayernormOpCPU(unittest.TestCase):
@@ -939,7 +939,7 @@ def test_residual_bias_add_layernorm(self):
 
 
 @unittest.skipIf(
-    not core.avx_supported() or not core.is_compiled_with_avx(),
+    not core.supports_avx512f() or not core.is_compiled_with_avx(),
     "machine is not support AVX or is not compiled with AVX",
 )
 class TestlayernormStaticOpCPU(unittest.TestCase):