Centralize cuda_arch capability definition

spack-environment/cuda/spack.yaml

@@ -2,18 +2,19 @@ spack:
   include:
   - ../concretizer.yaml
   - ../packages.yaml
+  - ../cuda_arch.yaml
   - ../packages_root_with_opengl.yaml
   - ../view.yaml
   concretizer:
     unify: when_possible # multiple epic versions
   specs:
-  - acts +cuda cuda_arch=75
+  - acts +cuda
   - actsvg
   - afterburner
-  - arrow +cuda cuda_arch=75
+  - arrow +cuda
   - cairo
   - catch2
-  - celeritas +cuda cuda_arch=75
+  - celeritas +cuda
   - cernlib
   - cli11
   - cmake
@@ -90,7 +91,7 @@ spack:
   - py-scipy
   - py-seaborn
   - py-toml
-  - py-torch +cuda cuda_arch=75
+  - py-torch +cuda
   - py-uproot
   - py-wurlitzer
   - py-yapf

spack-environment/cuda_arch.yaml

@@ -0,0 +1,6 @@
+# This file centralizes the CUDA architecture capability definition.
+# Current target: Compute Capability 7.5 (Turing: T4, RTX 2xxx, Quadro RTX)
+# To target a different GPU architecture, change cuda_arch here.
+packages:
+  all:
+    variants: cuda_arch=75

spack-environment/tf/spack.yaml

@@ -2,6 +2,7 @@ spack:
   include:
   - ../concretizer.yaml
   - ../packages.yaml
+  - ../cuda_arch.yaml
   - ../view.yaml
   concretizer:
     unify: when_possible  # py-numba needs a different llvm than the one provided as external
@@ -43,7 +44,7 @@ spack:
   - py-rucio-clients
   - py-scipy
   - py-seaborn
-  - py-tensorflow +cuda cuda_arch=75
+  - py-tensorflow +cuda
   - py-toml
   - py-uproot
   - py-vector