diff options
Diffstat (limited to 'share')
4 files changed, 50 insertions, 39 deletions
diff --git a/share/spack/gitlab/cloud_pipelines/stacks/ml-darwin-aarch64-mps/spack.yaml b/share/spack/gitlab/cloud_pipelines/stacks/ml-darwin-aarch64-mps/spack.yaml index 0526a4a12f..94211f6cde 100644 --- a/share/spack/gitlab/cloud_pipelines/stacks/ml-darwin-aarch64-mps/spack.yaml +++ b/share/spack/gitlab/cloud_pipelines/stacks/ml-darwin-aarch64-mps/spack.yaml @@ -3,8 +3,11 @@ spack: packages: all: - require: target=aarch64 - variants: +mps~cuda~rocm + require: + - target=aarch64 + - +mps + - ~cuda + - ~rocm mpi: require: openmpi openblas: diff --git a/share/spack/gitlab/cloud_pipelines/stacks/ml-linux-x86_64-cpu/spack.yaml b/share/spack/gitlab/cloud_pipelines/stacks/ml-linux-x86_64-cpu/spack.yaml index 32949a473f..ec28ec1fa3 100644 --- a/share/spack/gitlab/cloud_pipelines/stacks/ml-linux-x86_64-cpu/spack.yaml +++ b/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 - variants: ~cuda~rocm + require: + - target=x86_64_v3 + - ~cuda + - ~rocm mpi: require: openmpi diff --git a/share/spack/gitlab/cloud_pipelines/stacks/ml-linux-x86_64-cuda/spack.yaml b/share/spack/gitlab/cloud_pipelines/stacks/ml-linux-x86_64-cuda/spack.yaml index 22a56310ed..f9d74f6a4b 100644 --- a/share/spack/gitlab/cloud_pipelines/stacks/ml-linux-x86_64-cuda/spack.yaml +++ b/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 - variants: ~rocm+cuda cuda_arch=80 + require: + - target=x86_64_v3 + - ~rocm + - +cuda + - cuda_arch=80 llvm: # https://github.com/spack/spack/issues/27999 require: ~cuda diff --git a/share/spack/gitlab/cloud_pipelines/stacks/ml-linux-x86_64-rocm/spack.yaml b/share/spack/gitlab/cloud_pipelines/stacks/ml-linux-x86_64-rocm/spack.yaml index 9e546280f2..c83b4757fc 100644 --- a/share/spack/gitlab/cloud_pipelines/stacks/ml-linux-x86_64-rocm/spack.yaml +++ b/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 - variants: ~cuda+rocm amdgpu_target=gfx90a + require: + - 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-efficientnet-pytorch - # - py-gpytorch - # - py-kornia - # - py-lightning - # - py-pytorch-gradual-warmup-lr - # - py-pytorch-lightning - # - py-segmentation-models-pytorch - # - py-timm - # - py-torch - # - py-torch-cluster - # - py-torch-geometric - # - py-torch-nvidia-apex - # - py-torch-scatter - # - py-torch-sparse - # - py-torch-spline-conv - # - py-torchaudio - # - py-torchdata - # - py-torchfile - # - py-torchgeo - # - py-torchmetrics - # - py-torchtext - # - py-torchvision - # - py-vector-quantize-pytorch + - py-botorch + - py-efficientnet-pytorch + - py-gpytorch + - py-kornia + - py-lightning + - py-pytorch-gradual-warmup-lr + - py-pytorch-lightning + - py-segmentation-models-pytorch + - py-timm + - py-torch + - py-torch-cluster + - py-torch-geometric + - py-torch-nvidia-apex + - py-torch-scatter + - py-torch-sparse + - py-torch-spline-conv + - py-torchaudio + - py-torchdata + - py-torchfile + - py-torchgeo + - py-torchmetrics + - py-torchtext + - py-torchvision + - py-vector-quantize-pytorch # scikit-learn - py-scikit-learn @@ -72,12 +74,13 @@ spack: - py-tensorboardx # TensorFlow - - py-tensorflow - - py-tensorflow-datasets - - py-tensorflow-estimator - - py-tensorflow-hub - - py-tensorflow-metadata - - py-tensorflow-probability + # Does not yet support Spack-installed ROCm + # - py-tensorflow + # - py-tensorflow-datasets + # - py-tensorflow-estimator + # - py-tensorflow-hub + # - py-tensorflow-metadata + # - py-tensorflow-probability # XGBoost - py-xgboost |