diff --git a/python_bindings/setup.py b/python_bindings/setup.py index b6fe0b3..c6fa664 100755 --- a/python_bindings/setup.py +++ b/python_bindings/setup.py @@ -86,6 +86,10 @@ class BuildExt(build_ext): 'msvc': ['/EHsc', '/openmp', '/O2'], 'unix': ['-O3'], } + if 'ARCH' in os.environ: + # /arch:[IA32|SSE|SSE2|AVX|AVX2|ARMv7VE|VFPv4] + # See https://docs.microsoft.com/en-us/cpp/build/reference/arch-x86 + c_opts['msvc'].append("/arch:{}".format(os.envrion['ARCH'])) if 'CFLAGS' not in os.environ or "-march" not in os.environ["CFLAGS"]: c_opts['unix'].append('-march=native') link_opts = {