MXNet: add v1.9 (#36590)

2023-04-24 11:13:09 -05:00
parent 81d3df3289
commit b5d66e0144
4 changed files with 34 additions and 24 deletions
--- 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
@@ -24,6 +24,9 @@ spack:
    - py-keras-preprocessing
    - py-keras2onnx

+    # MXNet
+    - mxnet
+
    # PyTorch
    - py-botorch
    - py-efficientnet-pytorch
--- 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
@@ -27,6 +27,9 @@ spack:
    - py-keras-preprocessing
    - py-keras2onnx

+    # MXNet
+    - mxnet
+
    # PyTorch
    - py-botorch
    - py-efficientnet-pytorch
--- 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
@@ -29,6 +29,9 @@ spack:
    - py-keras-preprocessing
    - py-keras2onnx

+    # MXNet
+    - mxnet
+
    # PyTorch
    # Does not yet support Spack-install ROCm
    # - py-botorch