diff options
| author | Adam J. Stewart <ajstewart426@gmail.com> | 2023-08-08 03:59:24 -0500 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-08-08 10:59:24 +0200 |
| commit | 7327d2913a68ce4cf7d8bb07eff79c0f6415602e (patch) | |
| tree | af482a5413e2c255b23ccf15a1eea21b2a90a1c6 /share | |
| parent | 8f8a1f7f5223d0ac4e8b7731ad10611230633006 (diff) | |
| download | spack-7327d2913a68ce4cf7d8bb07eff79c0f6415602e.tar.gz spack-7327d2913a68ce4cf7d8bb07eff79c0f6415602e.tar.bz2 spack-7327d2913a68ce4cf7d8bb07eff79c0f6415602e.tar.xz spack-7327d2913a68ce4cf7d8bb07eff79c0f6415602e.zip | |
ML CI: get more packages working on macOS (#39199)
Diffstat (limited to 'share')
| -rw-r--r-- | share/spack/gitlab/cloud_pipelines/stacks/ml-darwin-aarch64-mps/spack.yaml | 51 |
1 files changed, 27 insertions, 24 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 7f22e232d1..504954df49 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 @@ -11,36 +11,52 @@ spack: require: ~fortran specs: + # Horovod + - py-horovod + # Hugging Face - py-transformers # JAX - # - py-jax # bazel codesign - # - py-jaxlib # bazel codesign + # Bazel codesign issues + # - py-jax + # - py-jaxlib # Keras + # Bazel codesign issues + # - py-keras - py-keras-applications - py-keras-preprocessing - py-keras2onnx - # - py-keras # bazel codesign # MXNet - mxnet # 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 @@ -53,32 +69,19 @@ spack: - py-tensorboardx # TensorFlow - # - py-tensorflow # bazel codesign - # - py-tensorflow-datasets # bazel codesign - # - py-tensorflow-hub # bazel codesign - # - py-tensorflow-metadata # bazel codesign - # - py-tensorflow-estimator # bazel codesign - # - py-tensorflow-probability # py-dm-tree due to bazel codesign + # Bazel codesign issues + # - py-tensorflow + # - py-tensorflow-datasets + # - py-tensorflow-estimator + # - py-tensorflow-hub + # - py-tensorflow-metadata + # - py-tensorflow-probability # XGBoost - py-xgboost + # - r-xgboost - xgboost - # ERRORS - # - py-efficientnet-pytorch # py-torch - # - py-horovod # py-torch - # - py-kornia # py-torch - # - py-lightning # py-torch - # - py-pytorch-lightning # py-torch - # - py-torch-nvidia-apex # py-torch - # - py-torch-scatter # py-torch - # - py-torch-spline-conv # py-torch - # - py-torchaudio # py-torchaudio - # - py-torchmetrics # py-torch - # - py-torchtext # py-torchtext - # - py-vector-quantize-pytorch # py-torch - # - r-xgboost # r - mirrors: { "mirror": "s3://spack-binaries/develop/ml-darwin-aarch64-mps" } ci: |