|
| 1 | +## |
| 2 | +# Copyright 2016-2021 Ghent University |
| 3 | +# Copyright 2016-2021 Forschungszentrum Juelich |
| 4 | +# |
| 5 | +# This file is part of EasyBuild, |
| 6 | +# originally created by the HPC team of Ghent University (http://ugent.be/hpc/en), |
| 7 | +# with support of Ghent University (http://ugent.be/hpc), |
| 8 | +# the Flemish Supercomputer Centre (VSC) (https://vscentrum.be/nl/en), |
| 9 | +# Flemish Research Foundation (FWO) (http://www.fwo.be/en) |
| 10 | +# and the Department of Economy, Science and Innovation (EWI) (http://www.ewi-vlaanderen.be/en). |
| 11 | +# |
| 12 | +# http://github.com/hpcugent/easybuild |
| 13 | +# |
| 14 | +# EasyBuild is free software: you can redistribute it and/or modify |
| 15 | +# it under the terms of the GNU General Public License as published by |
| 16 | +# the Free Software Foundation v2. |
| 17 | +# |
| 18 | +# EasyBuild is distributed in the hope that it will be useful, |
| 19 | +# but WITHOUT ANY WARRANTY; without even the implied warranty of |
| 20 | +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
| 21 | +# GNU General Public License for more details. |
| 22 | +# |
| 23 | +# You should have received a copy of the GNU General Public License |
| 24 | +# along with EasyBuild. If not, see <http://www.gnu.org/licenses/>. |
| 25 | +## |
| 26 | +""" |
| 27 | +EasyBuild support for nvompic compiler toolchain (includes NVHPC and OpenMPI, and CUDA as dependency). |
| 28 | +
|
| 29 | +:author: Damian Alvarez (Forschungszentrum Juelich) |
| 30 | +:author: Sebastian Achilles (Forschungszentrum Juelich) |
| 31 | +""" |
| 32 | + |
| 33 | +from easybuild.toolchains.nvhpc import NVHPCToolchain |
| 34 | +# We pull in MPI and CUDA at once so this maps nicely to HMNS |
| 35 | +from easybuild.toolchains.mpi.openmpi import OpenMPI |
| 36 | +from easybuild.toolchains.compiler.cuda import Cuda |
| 37 | + |
| 38 | + |
| 39 | +# Order matters! |
| 40 | +class NVompic(NVHPCToolchain, Cuda, OpenMPI): |
| 41 | + """Compiler toolchain with NVHPC and OpenMPI, with CUDA as dependency.""" |
| 42 | + NAME = 'nvompic' |
| 43 | + SUBTOOLCHAIN = NVHPCToolchain.NAME |
0 commit comments