77 lines
2 KiB
Python
77 lines
2 KiB
Python
"""
|
|
Auto-detect avx and xsave
|
|
|
|
According to Intel manual [0], both AVX and XSAVE features must be present to
|
|
use AVX instructions.
|
|
|
|
References:
|
|
|
|
[0] Intel Architecture Instruction Set Extensions Programming Reference
|
|
|
|
http://software.intel.com/sites/default/files/m/a/b/3/4/d/41604-319433-012a.pdf
|
|
|
|
"""
|
|
|
|
import sys
|
|
import os
|
|
import subprocess
|
|
import contextlib
|
|
|
|
|
|
def detect_avx_support(option='detect'):
|
|
'''Detect AVX support'''
|
|
option = os.environ.get('LLVMPY_AVX_SUPPORT', option).lower()
|
|
if option in ('disable', '0', 'false'):
|
|
return False
|
|
elif option in ('enable', '1', 'true'):
|
|
return True
|
|
else: # do detection
|
|
plat = sys.platform
|
|
if plat.startswith('darwin'):
|
|
return detect_osx_like()
|
|
elif plat.startswith('win32'):
|
|
return False # don't know how to detect in windows
|
|
else:
|
|
return detect_unix_like()
|
|
|
|
def detect_unix_like():
|
|
try:
|
|
info = open('/proc/cpuinfo')
|
|
except IOError:
|
|
return False
|
|
|
|
with contextlib.closing(info):
|
|
for line in info:
|
|
if line.lstrip().startswith('flags'):
|
|
features = line.split()
|
|
if 'avx' in features and 'xsave' in features:
|
|
# enable AVX if flags contain AVX
|
|
return True
|
|
return False
|
|
|
|
|
|
@contextlib.contextmanager
|
|
def _close_popen(popen):
|
|
if sys.version_info[0] >= 3:
|
|
with popen:
|
|
yield
|
|
else:
|
|
yield
|
|
popen.stdout.close()
|
|
|
|
|
|
def detect_osx_like():
|
|
try:
|
|
info = subprocess.Popen(['sysctl', '-n', 'machdep.cpu.features'],
|
|
stdout=subprocess.PIPE)
|
|
except OSError:
|
|
return False
|
|
|
|
with _close_popen(info):
|
|
features = info.stdout.read().decode('UTF8')
|
|
features = features.split()
|
|
return 'AVX1.0' in features and 'OSXSAVE' in features and 'XSAVE' in features
|
|
|
|
|
|
if __name__ == '__main__':
|
|
print("AVX support: %s" % detect_avx_support())
|