Move variant definitions for py-torch from spack-ext/repos/spack-stac…

…k/packages/ai-env/package.py to configs/common/packages.yaml
JCSDA · Mar 7, 2024 · 8424f72 · 8424f72
1 parent 59b7aa9
commit 8424f72
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/configs/common/packages.yaml b/configs/common/packages.yaml
@@ -242,6 +242,8 @@
       variants: +rust_bootstrap
     py-shapely:
       require: ['@1.8.0']
+    py-torch:
+      require: "+internal-protobuf ~mkldnn"
     qt:
       version: ['5.15.3']
     scotch:

diff --git a/spack-ext/repos/spack-stack/packages/ai-env/package.py b/spack-ext/repos/spack-stack/packages/ai-env/package.py
@@ -18,6 +18,6 @@ class AiEnv(BundlePackage):
     version("1.0.0")
 
     depends_on("jedi-base-env", type="run")
-    depends_on("py-torch +internal-protobuf", type="run")
+    depends_on("py-torch", type="run")
 
     # There is no need for install() since there is no code.