334 lines
12 KiB
Python
334 lines
12 KiB
Python
|
##############################################################################
|
||
|
|
# Copyright (c) 2013-2017, Lawrence Livermore National Security, LLC.
|
||
|
|
# Produced at the Lawrence Livermore National Laboratory.
|
||
|
|
#
|
||
|
|
# This file is part of Spack.
|
||
|
|
# Created by Todd Gamblin, tgamblin@llnl.gov, All rights reserved.
|
||
|
|
# LLNL-CODE-647188
|
||
|
|
#
|
||
|
|
# For details, see https://github.com/spack/spack
|
||
|
|
# Please also see the NOTICE and LICENSE files for our notice and the LGPL.
|
||
|
|
#
|
||
|
|
# This program is free software; you can redistribute it and/or modify
|
||
|
|
# it under the terms of the GNU Lesser General Public License (as
|
||
|
|
# published by the Free Software Foundation) version 2.1, February 1999.
|
||
|
|
#
|
||
|
|
# This program 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 terms and
|
||
|
|
# conditions of the GNU Lesser General Public License for more details.
|
||
|
|
#
|
||
|
|
# You should have received a copy of the GNU Lesser General Public
|
||
|
|
# License along with this program; if not, write to the Free Software
|
||
|
|
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
||
|
|
##############################################################################
|
||
|
|
|
||
|
|
|
||
|
|
from spack import *
|
||
|
|
|
||
|
|
import socket
|
||
|
|
import os
|
||
|
|
|
||
|
|
import llnl.util.tty as tty
|
||
|
|
from os import environ as env
|
||
|
|
|
||
|
|
|
||
|
|
def cmake_cache_entry(name, value):
|
||
|
|
"""
|
||
|
|
Helper that creates CMake cache entry strings used in
|
||
|
|
'host-config' files.
|
||
|
|
"""
|
||
|
|
return 'set({0} "{1}" CACHE PATH "")\n\n'.format(name, value)
|
||
|
|
|
||
|
|
|
||
|
|
class Ascent(Package):
|
||
|
|
"""Ascent is an open source many-core capable lightweight in situ
|
||
|
|
visualization and analysis infrastructure for multi-physics HPC
|
||
|
|
simulations."""
|
||
|
|
|
||
|
|
homepage = "https://github.com/Alpine-DAV/ascent"
|
||
|
|
url = "https://github.com/Alpine-DAV/ascent"
|
||
|
|
|
||
|
|
maintainers = ['cyrush']
|
||
|
|
|
||
|
|
version('develop',
|
||
|
|
git='https://github.com/Alpine-DAV/ascent.git',
|
||
|
|
branch='develop',
|
||
|
|
submodules=True)
|
||
|
|
|
||
|
|
###########################################################################
|
||
|
|
# package variants
|
||
|
|
###########################################################################
|
||
|
|
|
||
|
|
variant("shared", default=True, description="Build Conduit as shared libs")
|
||
|
|
|
||
|
|
variant("cmake", default=True,
|
||
|
|
description="Build CMake (if off, attempt to use cmake from PATH)")
|
||
|
|
|
||
|
|
variant("mpi", default=True, description="Build Ascent MPI Support")
|
||
|
|
|
||
|
|
# variants for python support
|
||
|
|
variant("python", default=True, description="Build Conduit Python support")
|
||
|
|
|
||
|
|
# variants for runtime features
|
||
|
|
|
||
|
|
variant("vtkh", default=True,
|
||
|
|
description="Build VTK-h filter and rendering support")
|
||
|
|
|
||
|
|
variant("tbb", default=True, description="Build tbb support")
|
||
|
|
variant("cuda", default=False, description="Build cuda support")
|
||
|
|
|
||
|
|
variant("adios", default=True, description="Build Adios filter support")
|
||
|
|
|
||
|
|
# variants for dev-tools (docs, etc)
|
||
|
|
variant("doc", default=False, description="Build Conduit's documentation")
|
||
|
|
|
||
|
|
###########################################################################
|
||
|
|
# package dependencies
|
||
|
|
###########################################################################
|
||
|
|
|
||
|
|
depends_on("cmake", when="+cmake")
|
||
|
|
depends_on("conduit")
|
||
|
|
|
||
|
|
#######################
|
||
|
|
# Python
|
||
|
|
#######################
|
||
|
|
# we need a shared version of python b/c linking with static python lib
|
||
|
|
# causes duplicate state issues when running compiled python modules.
|
||
|
|
depends_on("python+shared")
|
||
|
|
extends("python", when="+python")
|
||
|
|
# TODO: blas and lapack are disabled due to build
|
||
|
|
# issues Cyrus experienced on OSX 10.11.6
|
||
|
|
depends_on("py-numpy~blas~lapack", when="+python", type=('build', 'run'))
|
||
|
|
|
||
|
|
#######################
|
||
|
|
# MPI
|
||
|
|
#######################
|
||
|
|
depends_on("mpi", when="+mpi")
|
||
|
|
depends_on("py-mpi4py", when="+python+mpi")
|
||
|
|
|
||
|
|
#############################
|
||
|
|
# TPLs for Runtime Features
|
||
|
|
#############################
|
||
|
|
|
||
|
|
depends_on("vtkh", when="+vtkh")
|
||
|
|
depends_on("vtkh+cuda", when="+vtkh+cuda")
|
||
|
|
depends_on("adios", when="+adios")
|
||
|
|
|
||
|
|
#######################
|
||
|
|
# Documentation related
|
||
|
|
#######################
|
||
|
|
depends_on("py-sphinx", when="+python+doc", type='build')
|
||
|
|
|
||
|
|
def install(self, spec, prefix):
|
||
|
|
"""
|
||
|
|
Build and install Conduit.
|
||
|
|
"""
|
||
|
|
with working_dir('spack-build', create=True):
|
||
|
|
host_cfg_fname = self.create_host_config(spec, prefix)
|
||
|
|
cmake_args = []
|
||
|
|
# if we have a static build, we need to avoid any of
|
||
|
|
# spack's default cmake settings related to rpaths
|
||
|
|
# (see: https://github.com/LLNL/spack/issues/2658)
|
||
|
|
if "+shared" in spec:
|
||
|
|
cmake_args.extend(std_cmake_args)
|
||
|
|
else:
|
||
|
|
for arg in std_cmake_args:
|
||
|
|
if arg.count("RPATH") == 0:
|
||
|
|
cmake_args.append(arg)
|
||
|
|
cmake_args.extend(["-C", host_cfg_fname, "../src"])
|
||
|
|
cmake(*cmake_args)
|
||
|
|
make()
|
||
|
|
make("install")
|
||
|
|
# TODO also copy host_cfg_fname into install
|
||
|
|
|
||
|
|
def create_host_config(self, spec, prefix):
|
||
|
|
"""
|
||
|
|
This method creates a 'host-config' file that specifies
|
||
|
|
all of the options used to configure and build ascent.
|
||
|
|
"""
|
||
|
|
|
||
|
|
#######################
|
||
|
|
# Compiler Info
|
||
|
|
#######################
|
||
|
|
c_compiler = env["SPACK_CC"]
|
||
|
|
cpp_compiler = env["SPACK_CXX"]
|
||
|
|
f_compiler = None
|
||
|
|
|
||
|
|
if self.compiler.fc:
|
||
|
|
# even if this is set, it may not exist so do one more sanity check
|
||
|
|
if os.path.isfile(env["SPACK_FC"]):
|
||
|
|
f_compiler = env["SPACK_FC"]
|
||
|
|
|
||
|
|
#######################################################################
|
||
|
|
# By directly fetching the names of the actual compilers we appear
|
||
|
|
# to doing something evil here, but this is necessary to create a
|
||
|
|
# 'host config' file that works outside of the spack install env.
|
||
|
|
#######################################################################
|
||
|
|
|
||
|
|
sys_type = spec.architecture
|
||
|
|
# if on llnl systems, we can use the SYS_TYPE
|
||
|
|
if "SYS_TYPE" in env:
|
||
|
|
sys_type = env["SYS_TYPE"]
|
||
|
|
|
||
|
|
##############################################
|
||
|
|
# Find and record what CMake is used
|
||
|
|
##############################################
|
||
|
|
|
||
|
|
if "+cmake" in spec:
|
||
|
|
cmake_exe = spec['cmake'].command.path
|
||
|
|
else:
|
||
|
|
cmake_exe = which("cmake")
|
||
|
|
if cmake_exe is None:
|
||
|
|
msg = 'failed to find CMake (and cmake variant is off)'
|
||
|
|
raise RuntimeError(msg)
|
||
|
|
cmake_exe = cmake_exe.path
|
||
|
|
|
||
|
|
host_cfg_fname = "%s-%s-%s.cmake" % (socket.gethostname(),
|
||
|
|
sys_type,
|
||
|
|
spec.compiler)
|
||
|
|
|
||
|
|
cfg = open(host_cfg_fname, "w")
|
||
|
|
cfg.write("##################################\n")
|
||
|
|
cfg.write("# spack generated host-config\n")
|
||
|
|
cfg.write("##################################\n")
|
||
|
|
cfg.write("# {0}-{1}\n".format(sys_type, spec.compiler))
|
||
|
|
cfg.write("##################################\n\n")
|
||
|
|
|
||
|
|
# Include path to cmake for reference
|
||
|
|
cfg.write("# cmake from spack \n")
|
||
|
|
cfg.write("# cmake executable path: %s\n\n" % cmake_exe)
|
||
|
|
|
||
|
|
#######################
|
||
|
|
# Compiler Settings
|
||
|
|
#######################
|
||
|
|
|
||
|
|
cfg.write("#######\n")
|
||
|
|
cfg.write("# using %s compiler spec\n" % spec.compiler)
|
||
|
|
cfg.write("#######\n\n")
|
||
|
|
cfg.write("# c compiler used by spack\n")
|
||
|
|
cfg.write(cmake_cache_entry("CMAKE_C_COMPILER", c_compiler))
|
||
|
|
cfg.write("# cpp compiler used by spack\n")
|
||
|
|
cfg.write(cmake_cache_entry("CMAKE_CXX_COMPILER", cpp_compiler))
|
||
|
|
|
||
|
|
cfg.write("# fortran compiler used by spack\n")
|
||
|
|
if f_compiler is not None:
|
||
|
|
cfg.write(cmake_cache_entry("ENABLE_FORTRAN", "ON"))
|
||
|
|
cfg.write(cmake_cache_entry("CMAKE_Fortran_COMPILER", f_compiler))
|
||
|
|
else:
|
||
|
|
cfg.write("# no fortran compiler found\n\n")
|
||
|
|
cfg.write(cmake_cache_entry("ENABLE_FORTRAN", "OFF"))
|
||
|
|
|
||
|
|
#######################################################################
|
||
|
|
# Core Dependencies
|
||
|
|
#######################################################################
|
||
|
|
|
||
|
|
#######################
|
||
|
|
# Conduit
|
||
|
|
#######################
|
||
|
|
|
||
|
|
cfg.write("# conduit from spack \n")
|
||
|
|
cfg.write(cmake_cache_entry("CONDUIT_DIR", spec['conduit'].prefix))
|
||
|
|
|
||
|
|
#######################################################################
|
||
|
|
# Optional Dependencies
|
||
|
|
#######################################################################
|
||
|
|
|
||
|
|
#######################
|
||
|
|
# Python
|
||
|
|
#######################
|
||
|
|
|
||
|
|
cfg.write("# Python Support\n")
|
||
|
|
|
||
|
|
if "+python" in spec:
|
||
|
|
cfg.write("# Enable python module builds\n")
|
||
|
|
cfg.write(cmake_cache_entry("ENABLE_PYTHON", "ON"))
|
||
|
|
cfg.write("# python from spack \n")
|
||
|
|
cfg.write(cmake_cache_entry("PYTHON_EXECUTABLE",
|
||
|
|
spec['python'].command.path))
|
||
|
|
# install module to standard style site packages dir
|
||
|
|
# so we can support spack activate
|
||
|
|
cfg.write(cmake_cache_entry("PYTHON_MODULE_INSTALL_PREFIX",
|
||
|
|
site_packages_dir))
|
||
|
|
else:
|
||
|
|
cfg.write(cmake_cache_entry("ENABLE_PYTHON", "OFF"))
|
||
|
|
|
||
|
|
if "+doc" in spec:
|
||
|
|
cfg.write(cmake_cache_entry("ENABLE_DOCS", "ON"))
|
||
|
|
|
||
|
|
cfg.write("# sphinx from spack \n")
|
||
|
|
sphinx_build_exe = join_path(spec['py-sphinx'].prefix.bin,
|
||
|
|
"sphinx-build")
|
||
|
|
cfg.write(cmake_cache_entry("SPHINX_EXECUTABLE", sphinx_build_exe))
|
||
|
|
|
||
|
|
cfg.write("# doxygen from uberenv\n")
|
||
|
|
doxygen_exe = spec['doxygen'].command.path
|
||
|
|
cfg.write(cmake_cache_entry("DOXYGEN_EXECUTABLE", doxygen_exe))
|
||
|
|
else:
|
||
|
|
cfg.write(cmake_cache_entry("ENABLE_DOCS", "OFF"))
|
||
|
|
|
||
|
|
#######################
|
||
|
|
# MPI
|
||
|
|
#######################
|
||
|
|
|
||
|
|
cfg.write("# MPI Support\n")
|
||
|
|
|
||
|
|
if "+mpi" in spec:
|
||
|
|
cfg.write(cmake_cache_entry("ENABLE_MPI", "ON"))
|
||
|
|
cfg.write(cmake_cache_entry("MPI_C_COMPILER", spec['mpi'].mpicc))
|
||
|
|
cfg.write(cmake_cache_entry("MPI_CXX_COMPILER",
|
||
|
|
spec['mpi'].mpicxx))
|
||
|
|
cfg.write(cmake_cache_entry("MPI_Fortran_COMPILER",
|
||
|
|
spec['mpi'].mpifc))
|
||
|
|
else:
|
||
|
|
cfg.write(cmake_cache_entry("ENABLE_MPI", "OFF"))
|
||
|
|
|
||
|
|
#######################
|
||
|
|
# CUDA
|
||
|
|
#######################
|
||
|
|
|
||
|
|
cfg.write("# CUDA Support\n")
|
||
|
|
|
||
|
|
if "+cuda" in spec:
|
||
|
|
cfg.write(cmake_cache_entry("ENABLE_CUDA", "ON"))
|
||
|
|
else:
|
||
|
|
cfg.write(cmake_cache_entry("ENABLE_CUDA", "OFF"))
|
||
|
|
|
||
|
|
#######################
|
||
|
|
# VTK-h
|
||
|
|
#######################
|
||
|
|
|
||
|
|
cfg.write("# vtk-h support \n")
|
||
|
|
|
||
|
|
if "+vtkh" in spec:
|
||
|
|
cfg.write("# tbb from spack\n")
|
||
|
|
cfg.write(cmake_cache_entry("TBB_DIR", spec['tbb'].prefix))
|
||
|
|
|
||
|
|
cfg.write("# vtk-m from spack\n")
|
||
|
|
cfg.write(cmake_cache_entry("VTKM_DIR", spec['vtkm'].prefix))
|
||
|
|
|
||
|
|
cfg.write("# vtk-h from spack\n")
|
||
|
|
cfg.write(cmake_cache_entry("VTKH_DIR", spec['vtkh'].prefix))
|
||
|
|
else:
|
||
|
|
cfg.write("# vtk-h not built by spack \n")
|
||
|
|
|
||
|
|
#######################
|
||
|
|
# Adios
|
||
|
|
#######################
|
||
|
|
|
||
|
|
cfg.write("# adios support\n")
|
||
|
|
|
||
|
|
if "+adios" in spec:
|
||
|
|
cfg.write(cmake_cache_entry("ADIOS_DIR", spec['adios'].prefix))
|
||
|
|
else:
|
||
|
|
cfg.write("# adios not built by spack \n")
|
||
|
|
|
||
|
|
cfg.write("##################################\n")
|
||
|
|
cfg.write("# end spack generated host-config\n")
|
||
|
|
cfg.write("##################################\n")
|
||
|
|
cfg.close()
|
||
|
|
|
||
|
|
host_cfg_fname = os.path.abspath(host_cfg_fname)
|
||
|
|
tty.info("spack generated conduit host-config file: " + host_cfg_fname)
|
||
|
|
return host_cfg_fname
|