diff --git a/easybuild/easyblocks/m/mesa.py b/easybuild/easyblocks/m/mesa.py new file mode 100644 index 0000000000..cf8f299469 --- /dev/null +++ b/easybuild/easyblocks/m/mesa.py @@ -0,0 +1,122 @@ +## +# Copyright 2009-2020 Ghent University +# +# This file is part of EasyBuild, +# originally created by the HPC team of Ghent University (http://ugent.be/hpc/en), +# with support of Ghent University (http://ugent.be/hpc), +# the Flemish Supercomputer Centre (VSC) (https://www.vscentrum.be), +# Flemish Research Foundation (FWO) (http://www.fwo.be/en) +# and the Department of Economy, Science and Innovation (EWI) (http://www.ewi-vlaanderen.be/en). +# +# https://github.com/easybuilders/easybuild +# +# EasyBuild is free software: you can redistribute it and/or modify +# it under the terms of the GNU General Public License as published by +# the Free Software Foundation v2. +# +# EasyBuild is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU General Public License for more details. +# +# You should have received a copy of the GNU General Public License +# along with EasyBuild. If not, see . +## +""" +EasyBuild support for installing Mesa, implemented as an easyblock + +@author: Andrew Edmondson (University of Birmingham) +@author: Kenneth Hoste (HPC-UGent) +""" +import os +from distutils.version import LooseVersion + +from easybuild.easyblocks.generic.mesonninja import MesonNinja +from easybuild.tools.filetools import copy_dir +from easybuild.tools.systemtools import POWER, X86_64, get_cpu_architecture, get_cpu_features, get_shared_lib_ext + + +class EB_Mesa(MesonNinja): + """Custom easyblock for building and installing Mesa.""" + + def __init__(self, *args, **kwargs): + """Constructor for custom Mesa easyblock: figure out which vales to pass to swr-arches configuration option.""" + + super(EB_Mesa, self).__init__(*args, **kwargs) + + self.swr_arches = [] + + if 'swr-arches' not in self.cfg['configopts']: + # set cpu features of SWR for current architecture (only on x86_64) + if get_cpu_architecture() == X86_64: + feat_to_swrarch = { + 'avx': 'avx', + 'avx1.0': 'avx', # on macOS, AVX is indicated with 'avx1.0' rather than 'avx' + 'avx2': 'avx2', + 'avx512f': 'skx', # AVX-512 Foundation - introduced in Skylake + 'avx512er': 'knl', # AVX-512 Exponential and Reciprocal Instructions implemented in Knights Landing + } + # determine list of values to pass to swr-arches configuration option + cpu_features = get_cpu_features() + self.swr_arches = sorted([swrarch for feat, swrarch in feat_to_swrarch.items() if feat in cpu_features]) + + def configure_step(self): + """ + Customise the configure options based on the processor architecture of the host + (x86_64 or not, CPU features, ...) + """ + + arch = get_cpu_architecture() + if 'gallium-drivers' not in self.cfg['configopts']: + # Install appropriate Gallium drivers for current architecture + if arch == X86_64: + self.cfg.update('configopts', "-Dgallium-drivers='swrast,swr'") + elif arch == POWER: + self.cfg.update('configopts', "-Dgallium-drivers='swrast'") + + if self.swr_arches: + self.cfg.update('configopts', '-Dswr-arches=' + ','.join(self.swr_arches)) + + return super(EB_Mesa, self).configure_step() + + def install_step(self): + """Also copy additional header files after installing Mesa.""" + + super(EB_Mesa, self).install_step() + + # also install header files located in include/GL/internal, unless they're available already; + # we can't enable both DRI and Gallium drivers, + # but we can provide the DRI header file (GL/internal/dri_interface.h) + target_inc_GL_internal = os.path.join(self.installdir, 'include', 'GL', 'internal') + if not os.path.exists(target_inc_GL_internal): + src_inc_GL_internal = os.path.join(self.start_dir, 'include', 'GL', 'internal') + copy_dir(src_inc_GL_internal, target_inc_GL_internal) + self.log.info("Copied %s to %s" % (src_inc_GL_internal, target_inc_GL_internal)) + + def sanity_check_step(self): + """Custom sanity check for Mesa.""" + + shlib_ext = get_shared_lib_ext() + + if LooseVersion(self.version) >= LooseVersion('20.0'): + header_files = [os.path.join('include', 'EGL', x) for x in ['eglmesaext.h', 'eglextchromium.h']] + header_files.extend([ + os.path.join('include', 'GL', 'osmesa.h'), + os.path.join('include', 'GL', 'internal', 'dri_interface.h'), + ]) + else: + gl_inc_files = ['glext.h', 'gl_mangle.h', 'glx.h', 'osmesa.h', 'gl.h', 'glxext.h', 'glx_mangle.h'] + gles_inc_files = [('GLES', 'gl.h'), ('GLES2', 'gl2.h'), ('GLES3', 'gl3.h')] + header_files = [os.path.join('include', 'GL', x) for x in gl_inc_files] + header_files.extend([os.path.join('include', x, y) for (x, y) in gles_inc_files]) + + custom_paths = { + 'files': [os.path.join('lib', 'libOSMesa.%s' % shlib_ext)] + header_files, + 'dirs': [os.path.join('include', 'GL', 'internal')], + } + + if self.swr_arches: + swr_arch_libs = [os.path.join('lib', 'libswr%s.%s' % (a.upper(), shlib_ext)) for a in self.swr_arches] + custom_paths['files'].extend(swr_arch_libs) + + super(EB_Mesa, self).sanity_check_step(custom_paths=custom_paths)