.pre-commit-config.yaml

@@ -21,9 +21,3 @@ repos:
   rev: v0.18
   hooks:
     - id: validate-pyproject
-
-- repo: https://github.com/src-r-r/pre-commit-license-year
-  rev: bde06feb9eeb1438a6d1d80b2471e668dd34162b
-  hooks:
-    - id: check_year
-      name: check LICENSE year

src/mpsd_software_manager/spack.py

@@ -351,7 +351,11 @@ def install_package_set_from_environment(spack_environment: str) -> None:
         logger.debug(spack(f"env create {spack_environment}"))
 
     config = Config()
-    context = {"toolchain_compiler": compilers["default"]["name"]}
+    context = {
+        "toolchain_compiler": compilers["default"]["name"],
+        # cuda arch: avocado (61), gpu (70), gpu-ayyer (70), accelerated tentacles (75)
+        "cuda_arch": "61,70,75",
+    }
     if "fallback" in compilers:
         context["fallback_compiler"] = compilers["fallback"]["name"]
     render_template(
@@ -634,11 +638,9 @@ def install_toolchain_compiler(spack_environment: str) -> dict[str, dict[str, st
         gcc = compilers["fallback"]
         intel = compilers["default"]
         arch_tag = platform.machine()
-        cflags = f" -idirafter/usr/include/{arch_tag}-linux-gnu" " -diag-disable=10441"
-        cxxflags = (
-            f" -idirafter/usr/include/{arch_tag}-linux-gnu" " -diag-disable=10441"
-        )
-        fflags = f" -idirafter/usr/include/{arch_tag}-linux-gnu" " -diag-disable=10441"
+        cflags = f" -idirafter/usr/include/{arch_tag}-linux-gnu -diag-disable=10441"
+        cxxflags = f" -idirafter/usr/include/{arch_tag}-linux-gnu -diag-disable=10441"
+        fflags = f" -idirafter/usr/include/{arch_tag}-linux-gnu -diag-disable=10441"
 
         intel_spec = intel["name"].replace("@", "@=")