Skip to content
Snippets Groups Projects
Select Git revision
  • a0e0cbbf8e5648170137a78dac95d6435a1a0c7f
  • master default protected
  • pytorch-2.0
3 results

utils.py

Blame
  • npsmpic.py 1.65 KiB
    ##
    # Copyright 2016-2016 Ghent University
    # Copyright 2016-2016 Forschungszentrum Juelich
    #
    # 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://vscentrum.be/nl/en),
    # Flemish Research Foundation (FWO) (http://www.fwo.be/en)
    # and the Department of Economy, Science and Innovation (EWI) (http://www.ewi-vlaanderen.be/en).
    #
    # http://github.com/hpcugent/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 <http://www.gnu.org/licenses/>.
    ##
    """
    EasyBuild support for npsmpi compiler toolchain (includes NVHPC and ParaStationMPI, and CUDA as dependency).
    
    @author: Damian Alvarez (Forschungszentrum Juelich)
    """
    
    from easybuild.toolchains.nvhpc import NVHPCToolchain
    # We pull in MPI and CUDA at once so this maps nicely to HMNS
    from easybuild.toolchains.mpi.psmpi import Psmpi
    from easybuild.toolchains.compiler.cuda import Cuda
    
    # Order matters!
    
    
    class Npsmpic(NVHPCToolchain, Cuda, Psmpi):
        """Compiler toolchain with NVHPC and ParaStationMPI, with CUDA as dependency."""
        NAME = 'npsmpic'
        SUBTOOLCHAIN = NVHPCToolchain.NAME