Commit f8b8a718 authored by one's avatar one
Browse files

Format

parent 52f4786f
......@@ -4,13 +4,12 @@
import os
from spack.package import *
from spack_repo.builtin.build_systems import cmake
from spack_repo.builtin.build_systems.cmake import CMakePackage
from spack_repo.builtin.build_systems.cuda import CudaPackage
from spack_repo.builtin.build_systems.rocm import ROCmPackage
from spack.package import *
class Gromacs(CMakePackage, CudaPackage, ROCmPackage):
"""GROMACS is a molecular dynamics package primarily designed for simulations
......@@ -604,7 +603,9 @@ class CMakeBuilder(cmake.CMakeBuilder):
options.append("-DGMX_GPU:STRING=SYCL")
if self.spec.satisfies("^hipsycl") or self.spec.satisfies("^adaptivecpp"):
options.append("-DGMX_SYCL=ACPP")
options.append(f"-DACPP_TARGETS=hip:{self.spec.variants['amdgpu_target'].value}")
options.append(
f"-DACPP_TARGETS=hip:{self.spec.variants['amdgpu_target'].value}"
)
else:
options.append("-DGMX_GPU:STRING=OFF")
else:
......
......@@ -5,11 +5,10 @@
import json
import os
from spack.package import *
from spack_repo.builtin.build_systems.cmake import CMakePackage
from spack_repo.builtin.build_systems.rocm import ROCmPackage
from spack.package import *
class Hipsycl(CMakePackage, ROCmPackage):
"""hipSYCL is an implementation of the SYCL standard programming model
......@@ -148,7 +147,9 @@ class Hipsycl(CMakePackage, ROCmPackage):
args.append("-DROCM_PATH:STRING={0}".format(os.environ.get("ROCM_PATH")))
# Work around HIP_CLANG_INCLUDE_PATH-NOTFOUND
args.append(
"-DHIP_CLANG_INCLUDE_PATH:String={0}".format(os.path.dirname(llvm_clang_include_dirs[0]))
"-DHIP_CLANG_INCLUDE_PATH:String={0}".format(
os.path.dirname(llvm_clang_include_dirs[0])
)
)
if self.spec.satisfies("@24.02.0:"):
args.append("-DWITH_SSCP_COMPILER=OFF")
......
......@@ -3,12 +3,11 @@
# SPDX-License-Identifier: (Apache-2.0 OR MIT)
import os
from spack.package import *
from spack_repo.builtin.build_systems.cmake import CMakePackage
from spack_repo.builtin.build_systems.cuda import CudaPackage
from spack_repo.builtin.build_systems.rocm import ROCmPackage
from spack.package import *
class Kokkos(CMakePackage, CudaPackage, ROCmPackage):
"""Kokkos implements a programming model in C++ for writing performance
......
......@@ -4,14 +4,13 @@
import os
from spack.package import *
from spack_repo.builtin.build_systems import cmake, generic
from spack_repo.builtin.build_systems.cmake import CMakePackage
from spack_repo.builtin.build_systems.cuda import CudaPackage
from spack_repo.builtin.build_systems.generic import Package
from spack_repo.builtin.build_systems.rocm import ROCmPackage
from spack.package import *
class Nekrs(Package, CMakePackage, CudaPackage, ROCmPackage):
"""nekRS is an open-source Navier Stokes solver based on the spectral
......@@ -52,7 +51,12 @@ class Nekrs(Package, CMakePackage, CudaPackage, ROCmPackage):
variant("opencl", default=False, description="Activates support for OpenCL")
variant("gpu_aware_mpi", default=False, description="Enable GPU-aware MPI support")
variant("adios", default=False, when="@26.0:", description="Enable ADIOS2 checkpoint engine")
variant("cpptrace", default=False, when="@26.0:", description="Enable stack tracing via bundled cpptrace")
variant(
"cpptrace",
default=False,
when="@26.0:",
description="Enable stack tracing via bundled cpptrace",
)
variant(
"single_precision",
default=True,
......@@ -112,7 +116,7 @@ class Nekrs(Package, CMakePackage, CudaPackage, ROCmPackage):
filter_file(
r"file\(MAKE_DIRECTORY \$\{CMAKE_INSTALL_PREFIX\}/3rd_party\)",
"install(FILES ${CMAKE_CURRENT_BINARY_DIR}/nekrs.conf DESTINATION .)\n"
"install(CODE \"file(MAKE_DIRECTORY \\\"${CMAKE_INSTALL_PREFIX}/3rd_party\\\")\")",
'install(CODE "file(MAKE_DIRECTORY \\"${CMAKE_INSTALL_PREFIX}/3rd_party\\")")',
"CMakeLists.txt",
)
......@@ -170,6 +174,7 @@ class Nekrs(Package, CMakePackage, CudaPackage, ROCmPackage):
cuda_dir = spec["cuda"].prefix
env.set("OCCA_CUDA_COMPILER", join_path(cuda_dir, "bin", "nvcc"))
class GenericBuilder(generic.GenericBuilder):
def install(self, pkg, spec, prefix):
makenrs = Executable(os.path.join(os.getcwd(), "makenrs"))
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment