summaryrefslogtreecommitdiff
path: root/share/spack/gitlab/cloud_pipelines/stacks/ml-darwin-aarch64-mps/spack.yaml
blob: 0905305113f083baed184a95878cdcb509feaf6a (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
spack:
  view: false

  packages:
    all:
      require: target=aarch64
      variants: +mps~cuda~rocm
    mpi:
      require: openmpi
    openblas:
      require: ~fortran

  specs:
  # Horovod
  - py-horovod

  # Hugging Face
  - py-transformers

  # JAX
  # Bazel codesign issues
  # - py-jax
  # - py-jaxlib

  # Keras
  # Bazel codesign issues
  # - py-keras
  - py-keras-applications
  - py-keras-preprocessing
  - py-keras2onnx

  # MXNet not supported on darwin aarch64 yet
  # - 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
  - py-scikit-learn-extra

  # TensorBoard
  - py-tensorboard
  - py-tensorboard-data-server
  - py-tensorboard-plugin-wit
  - py-tensorboardx

  # TensorFlow
  # 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

  ci:
    pipeline-gen:
    - build-job-remove:
        tags: [ spack, public ]
    - build-job:
        variables:
          CI_GPG_KEY_ROOT: /etc/protected-runner
        tags: [ "macos-ventura", "apple-clang-15", "aarch64-macos" ]

  cdash:
    build-group: Machine Learning MPS
    url: https://cdash.spack.io
    project: Spack Testing
    site: Cloud Gitlab Infrastructure