summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorAdam J. Stewart <ajstewart426@gmail.com>2022-10-10 13:18:05 -0500
committerGitHub <noreply@github.com>2022-10-10 12:18:05 -0600
commit831d7979caefed9dfe1051bab72d5babfba92242 (patch)
treefa1ae1b2993e16abfe9545a5a88648679ac368b6
parent14f6de9bf2df9742cb20a881c4d85ee85f7091f5 (diff)
downloadspack-831d7979caefed9dfe1051bab72d5babfba92242.tar.gz
spack-831d7979caefed9dfe1051bab72d5babfba92242.tar.bz2
spack-831d7979caefed9dfe1051bab72d5babfba92242.tar.xz
spack-831d7979caefed9dfe1051bab72d5babfba92242.zip
ML CPU pipeline: test py-torch-nvidia-apex (#33158)
-rw-r--r--share/spack/gitlab/cloud_pipelines/stacks/ml-cpu/spack.yaml3
1 files changed, 1 insertions, 2 deletions
diff --git a/share/spack/gitlab/cloud_pipelines/stacks/ml-cpu/spack.yaml b/share/spack/gitlab/cloud_pipelines/stacks/ml-cpu/spack.yaml
index a687a6928e..e6b49b9f41 100644
--- a/share/spack/gitlab/cloud_pipelines/stacks/ml-cpu/spack.yaml
+++ b/share/spack/gitlab/cloud_pipelines/stacks/ml-cpu/spack.yaml
@@ -46,8 +46,7 @@ spack:
- py-torch
- py-torch-cluster
- py-torch-geometric
- # https://github.com/NVIDIA/apex/issues/1498
- # - py-torch-nvidia-apex
+ - py-torch-nvidia-apex
- py-torch-scatter
- py-torch-sparse
- py-torch-spline-conv