cmrit
/
cmrithackathon-master
/.venv
/lib
/python3.11
/site-packages
/numpy
/distutils
/command
/build.py
import os | |
import sys | |
from distutils.command.build import build as old_build | |
from distutils.util import get_platform | |
from numpy.distutils.command.config_compiler import show_fortran_compilers | |
class build(old_build): | |
sub_commands = [('config_cc', lambda *args: True), | |
('config_fc', lambda *args: True), | |
('build_src', old_build.has_ext_modules), | |
] + old_build.sub_commands | |
user_options = old_build.user_options + [ | |
('fcompiler=', None, | |
"specify the Fortran compiler type"), | |
('warn-error', None, | |
"turn all warnings into errors (-Werror)"), | |
('cpu-baseline=', None, | |
"specify a list of enabled baseline CPU optimizations"), | |
('cpu-dispatch=', None, | |
"specify a list of dispatched CPU optimizations"), | |
('disable-optimization', None, | |
"disable CPU optimized code(dispatch,simd,fast...)"), | |
('simd-test=', None, | |
"specify a list of CPU optimizations to be tested against NumPy SIMD interface"), | |
] | |
help_options = old_build.help_options + [ | |
('help-fcompiler', None, "list available Fortran compilers", | |
show_fortran_compilers), | |
] | |
def initialize_options(self): | |
old_build.initialize_options(self) | |
self.fcompiler = None | |
self.warn_error = False | |
self.cpu_baseline = "min" | |
self.cpu_dispatch = "max -xop -fma4" # drop AMD legacy features by default | |
self.disable_optimization = False | |
""" | |
the '_simd' module is a very large. Adding more dispatched features | |
will increase binary size and compile time. By default we minimize | |
the targeted features to those most commonly used by the NumPy SIMD interface(NPYV), | |
NOTE: any specified features will be ignored if they're: | |
- part of the baseline(--cpu-baseline) | |
- not part of dispatch-able features(--cpu-dispatch) | |
- not supported by compiler or platform | |
""" | |
self.simd_test = "BASELINE SSE2 SSE42 XOP FMA4 (FMA3 AVX2) AVX512F " \ | |
"AVX512_SKX VSX VSX2 VSX3 VSX4 NEON ASIMD VX VXE VXE2" | |
def finalize_options(self): | |
build_scripts = self.build_scripts | |
old_build.finalize_options(self) | |
plat_specifier = ".{}-{}.{}".format(get_platform(), *sys.version_info[:2]) | |
if build_scripts is None: | |
self.build_scripts = os.path.join(self.build_base, | |
'scripts' + plat_specifier) | |
def run(self): | |
old_build.run(self) | |