ML CI: variants are now required (#42851)

share/spack/gitlab/cloud_pipelines/stacks/ml-darwin-aarch64-mps/spack.yaml

@@ -3,8 +3,11 @@ spack:
 
   packages:
     all:
-      require: target=aarch64
+      require:
-      variants: +mps~cuda~rocm
+      - target=aarch64
+      - +mps
+      - ~cuda
+      - ~rocm
     mpi:
       require: openmpi
     openblas:

share/spack/gitlab/cloud_pipelines/stacks/ml-linux-x86_64-cpu/spack.yaml

@@ -2,8 +2,10 @@ spack:
   view: false
   packages:
     all:
-      require: target=x86_64_v3
+      require:
-      variants: ~cuda~rocm
+      - target=x86_64_v3
+      - ~cuda
+      - ~rocm
     mpi:
       require: openmpi
 

share/spack/gitlab/cloud_pipelines/stacks/ml-linux-x86_64-cuda/spack.yaml

@@ -2,8 +2,11 @@ spack:
   view: false
   packages:
     all:
-      require: target=x86_64_v3
+      require:
-      variants: ~rocm+cuda cuda_arch=80
+      - target=x86_64_v3
+      - ~rocm
+      - +cuda
+      - cuda_arch=80
     llvm:
       # https://github.com/spack/spack/issues/27999
       require: ~cuda

share/spack/gitlab/cloud_pipelines/stacks/ml-linux-x86_64-rocm/spack.yaml

@@ -2,8 +2,11 @@ spack:
   view: false
   packages:
     all:
-      require: target=x86_64_v3
+      require:
-      variants: ~cuda+rocm amdgpu_target=gfx90a
+      - target=x86_64_v3
+      - ~cuda
+      - +rocm
+      - amdgpu_target=gfx90a
     gl:
       require: "osmesa"
     py-torch:
@@ -35,31 +38,30 @@ spack:
     - mxnet
 
     # PyTorch
-    # Does not yet support Spack-install ROCm
+    - py-botorch
-    # - py-botorch
+    - py-efficientnet-pytorch
-    # - py-efficientnet-pytorch
+    - py-gpytorch
-    # - py-gpytorch
+    - py-kornia
-    # - py-kornia
+    - py-lightning
-    # - py-lightning
+    - py-pytorch-gradual-warmup-lr
-    # - py-pytorch-gradual-warmup-lr
+    - py-pytorch-lightning
-    # - py-pytorch-lightning
+    - py-segmentation-models-pytorch
-    # - py-segmentation-models-pytorch
+    - py-timm
-    # - py-timm
+    - py-torch
-    # - py-torch
+    - py-torch-cluster
-    # - py-torch-cluster
+    - py-torch-geometric
-    # - py-torch-geometric
+    - py-torch-nvidia-apex
-    # - py-torch-nvidia-apex
+    - py-torch-scatter
-    # - py-torch-scatter
+    - py-torch-sparse
-    # - py-torch-sparse
+    - py-torch-spline-conv
-    # - py-torch-spline-conv
+    - py-torchaudio
-    # - py-torchaudio
+    - py-torchdata
-    # - py-torchdata
+    - py-torchfile
-    # - py-torchfile
+    - py-torchgeo
-    # - py-torchgeo
+    - py-torchmetrics
-    # - py-torchmetrics
+    - py-torchtext
-    # - py-torchtext
+    - py-torchvision
-    # - py-torchvision
+    - py-vector-quantize-pytorch
-    # - py-vector-quantize-pytorch
 
     # scikit-learn
     - py-scikit-learn
@@ -72,12 +74,13 @@ spack:
     - py-tensorboardx
 
     # TensorFlow
-    - py-tensorflow
+    # Does not yet support Spack-installed ROCm
-    - py-tensorflow-datasets
+    # - py-tensorflow
-    - py-tensorflow-estimator
+    # - py-tensorflow-datasets
-    - py-tensorflow-hub
+    # - py-tensorflow-estimator
-    - py-tensorflow-metadata
+    # - py-tensorflow-hub
-    - py-tensorflow-probability
+    # - py-tensorflow-metadata
+    # - py-tensorflow-probability
 
     # XGBoost
     - py-xgboost