core remove pycpuinfo (#18479)

remove pycpuinfo deps in core

Dockerfile

@@ -92,17 +92,17 @@ RUN localedef -i en_US -f UTF-8 en_US.UTF-8
 # specify sphinx version as 1.5.6 and remove -U option for [pip install -U
 # sphinx-rtd-theme] since -U option will cause sphinx being updated to newest
 # version(1.7.1 for now), which causes building documentation failed.
-RUN pip3 --no-cache-dir install -U wheel py-cpuinfo==5.0.0 && \
+RUN pip3 --no-cache-dir install -U wheel && \
     pip3 --no-cache-dir install -U docopt PyYAML sphinx==1.5.6 && \
     pip3 --no-cache-dir install sphinx-rtd-theme==0.1.9 recommonmark && \
-    pip3.6 --no-cache-dir install -U wheel py-cpuinfo==5.0.0 && \
+    pip3.6 --no-cache-dir install -U wheel && \
     pip3.6 --no-cache-dir install -U docopt PyYAML sphinx==1.5.6 && \
     pip3.6 --no-cache-dir install sphinx-rtd-theme==0.1.9 recommonmark && \
-    pip3.7 --no-cache-dir install -U wheel py-cpuinfo==5.0.0 && \
+    pip3.7 --no-cache-dir install -U wheel && \
     pip3.7 --no-cache-dir install -U docopt PyYAML sphinx==1.5.6 && \
     pip3.7 --no-cache-dir install sphinx-rtd-theme==0.1.9 recommonmark && \
     easy_install -U pip && \
-    pip --no-cache-dir install -U pip setuptools wheel py-cpuinfo==5.0.0 && \
+    pip --no-cache-dir install -U pip setuptools wheel && \
     pip --no-cache-dir install -U docopt PyYAML sphinx==1.5.6 && \
     pip --no-cache-dir install sphinx-rtd-theme==0.1.9 recommonmark
 

paddle/scripts/Dockerfile.tmp

@@ -92,17 +92,17 @@ RUN localedef -i en_US -f UTF-8 en_US.UTF-8
 # specify sphinx version as 1.5.6 and remove -U option for [pip install -U
 # sphinx-rtd-theme] since -U option will cause sphinx being updated to newest
 # version(1.7.1 for now), which causes building documentation failed.
-RUN pip3 --no-cache-dir install -U wheel py-cpuinfo==5.0.0 && \
+RUN pip3 --no-cache-dir install -U wheel && \
     pip3 --no-cache-dir install -U docopt PyYAML sphinx==1.5.6 && \
     pip3 --no-cache-dir install sphinx-rtd-theme==0.1.9 recommonmark && \
-    pip3.6 --no-cache-dir install -U wheel py-cpuinfo==5.0.0 && \
+    pip3.6 --no-cache-dir install -U wheel && \
     pip3.6 --no-cache-dir install -U docopt PyYAML sphinx==1.5.6 && \
     pip3.6 --no-cache-dir install sphinx-rtd-theme==0.1.9 recommonmark && \
-    pip3.7 --no-cache-dir install -U wheel py-cpuinfo==5.0.0 && \
+    pip3.7 --no-cache-dir install -U wheel && \
     pip3.7 --no-cache-dir install -U docopt PyYAML sphinx==1.5.6 && \
     pip3.7 --no-cache-dir install sphinx-rtd-theme==0.1.9 recommonmark && \
     easy_install -U pip && \
-    pip --no-cache-dir install -U pip setuptools wheel py-cpuinfo==5.0.0 && \
+    pip --no-cache-dir install -U pip setuptools wheel && \
     pip --no-cache-dir install -U docopt PyYAML sphinx==1.5.6 && \
     pip --no-cache-dir install sphinx-rtd-theme==0.1.9 recommonmark
 

python/paddle/fluid/core.py

@@ -16,7 +16,6 @@ from __future__ import print_function
 
 import sys
 import os
-from cpuinfo import get_cpu_info
 
 core_suffix = 'so'
 if os.name == 'nt':
@@ -57,27 +56,112 @@ except ImportError as e:
 except Exception as e:
     raise e
 
-load_noavx = False
 
-has_avx = False
-if sys.platform == 'darwin':
+def avx_supported():
+    """
+    Whether current system(Linux, MacOS, Windows) is supported with AVX.
+    """
+    import platform
+    from .. import compat as cpt
+    sysstr = platform.system().lower()
+    has_avx = False
+    if sysstr == 'linux':
         try:
-        has_avx = os.popen('sysctl machdep.cpu.features | grep -i avx').read(
-        ) != ''
+            has_avx = os.popen('cat /proc/cpuinfo | grep -i avx').read() != ''
+        except Exception as e:
+            sys.stderr.write('Can not get the AVX flag from /proc/cpuinfo.\n'
+                             'The original error is: %s\n' %
+                             cpt.get_exception_message(e))
+        return has_avx
+    elif sysstr == 'darwin':
+        try:
+            has_avx = os.popen(
+                'sysctl machdep.cpu.features | grep -i avx').read() != ''
         except Exception as e:
             sys.stderr.write(
-            'Can not get the AVX flag from machdep.cpu.features.\n')
+                'Can not get the AVX flag from machdep.cpu.features.\n'
+                'The original error is: %s\n' % cpt.get_exception_message(e))
         if not has_avx:
             try:
                 has_avx = os.popen(
-                'sysctl machdep.cpu.leaf7_features | grep -i avx').read() != ''
+                    'sysctl machdep.cpu.leaf7_features | grep -i avx').read(
+                    ) != ''
             except Exception as e:
                 sys.stderr.write(
-                'Can not get the AVX flag from machdep.cpu.leaf7_features.\n')
-else:
-    has_avx = 'avx' in get_cpu_info()['flags']
+                    'Can not get the AVX flag from machdep.cpu.leaf7_features.\n'
+                    'The original error is: %s\n' %
+                    cpt.get_exception_message(e))
+        return has_avx
+    elif sysstr == 'windows':
+        import ctypes
+        ONE_PAGE = ctypes.c_size_t(0x1000)
+
+        def asm_func(code_str, restype=ctypes.c_uint32, argtypes=()):
+            # Call the code_str as a function
+            # Alloc 1 page to ensure the protection
+            pfnVirtualAlloc = ctypes.windll.kernel32.VirtualAlloc
+            pfnVirtualAlloc.restype = ctypes.c_void_p
+            MEM_COMMIT = ctypes.c_ulong(0x1000)
+            PAGE_READWRITE = ctypes.c_ulong(0x4)
+            address = pfnVirtualAlloc(None, ONE_PAGE, MEM_COMMIT,
+                                      PAGE_READWRITE)
+            if not address:
+                raise Exception("Failed to VirtualAlloc")
+
+            # Copy the code into the memory segment
+            memmove = ctypes.CFUNCTYPE(ctypes.c_void_p, ctypes.c_void_p,
+                                       ctypes.c_void_p,
+                                       ctypes.c_size_t)(ctypes._memmove_addr)
+            if memmove(address, code_str, len(code_str)) < 0:
+                raise Exception("Failed to memmove")
+
+            # Enable execute permissions
+            PAGE_EXECUTE = ctypes.c_ulong(0x10)
+            pfnVirtualProtect = ctypes.windll.kernel32.VirtualProtect
+            res = pfnVirtualProtect(
+                ctypes.c_void_p(address), ONE_PAGE, PAGE_EXECUTE,
+                ctypes.byref(ctypes.c_ulong(0)))
+            if not res:
+                raise Exception("Failed VirtualProtect")
 
-if has_avx:
+            # Flush instruction cache
+            pfnGetCurrentProcess = ctypes.windll.kernel32.GetCurrentProcess
+            pfnGetCurrentProcess.restype = ctypes.c_void_p
+            prochandle = ctypes.c_void_p(pfnGetCurrentProcess())
+            res = ctypes.windll.kernel32.FlushInstructionCache(
+                prochandle, ctypes.c_void_p(address), ONE_PAGE)
+            if not res:
+                raise Exception("Failed FlushInstructionCache")
+
+            # Cast the memory to function
+            functype = ctypes.CFUNCTYPE(restype, *argtypes)
+            func = functype(address)
+            return func, address
+
+        # http://en.wikipedia.org/wiki/CPUID#EAX.3D1:_Processor_Info_and_Feature_Bits
+        # mov eax,0x1; cpuid; mov cx, ax; ret
+        code_str = b"\xB8\x01\x00\x00\x00\x0f\xa2\x89\xC8\xC3"
+        avx_bit = 28
+        retval = 0
+        try:
+            # Convert the code_str into a function that returns uint
+            func, address = asm_func(code_str)
+            retval = func()
+            ctypes.windll.kernel32.VirtualFree(
+                ctypes.c_void_p(address), ctypes.c_size_t(0), ONE_PAGE)
+        except Exception as e:
+            sys.stderr.write('Failed getting the AVX flag on Windows.\n'
+                             'The original error is: %s\n' %
+                             cpt.get_exception_message(e))
+        return (retval & (1 << avx_bit)) > 0
+    else:
+        sys.stderr.write('Do not get AVX flag on %s\n' % sysstr)
+        return False
+
+
+load_noavx = False
+
+if avx_supported():
     try:
         from .core_avx import *
         from .core_avx import __doc__, __file__, __name__, __package__
@@ -91,17 +175,17 @@ if has_avx:
         from .core_avx import _set_fuse_parameter_memory_size
         from .core_avx import _is_dygraph_debug_enabled
         from .core_avx import _dygraph_debug_level
-    except ImportError as e:
+    except Exception as e:
         if has_avx_core:
             raise e
         else:
+            from .. import compat as cpt
             sys.stderr.write(
                 'WARNING: Do not have avx core. You may not build with AVX, '
                 'but AVX is supported on local machine.\n You could build paddle '
-                'WITH_AVX=ON to get better performance.\n')
+                'WITH_AVX=ON to get better performance.\n'
+                'The original error is: %s\n' % cpt.get_exception_message(e))
             load_noavx = True
-    except Exception as e:
-        raise e
 else:
     load_noavx = True
 
@@ -119,11 +203,9 @@ if load_noavx:
         from .core_noavx import _set_fuse_parameter_memory_size
         from .core_noavx import _is_dygraph_debug_enabled
         from .core_noavx import _dygraph_debug_level
-    except ImportError as e:
+    except Exception as e:
         if has_noavx_core:
             sys.stderr.write(
                 'Error: Can not import noavx core while this file exists ' +
                 current_path + os.sep + 'core_noavx.' + core_suffix + '\n')
         raise e
-    except Exception as e:
-        raise e

python/requirements.txt

@@ -16,4 +16,3 @@ funcsigs
 pyyaml
 decorator
 prettytable
-py-cpuinfo==5.0.0