From 831d7979caefed9dfe1051bab72d5babfba92242 Mon Sep 17 00:00:00 2001 From: "Adam J. Stewart" Date: Mon, 10 Oct 2022 13:18:05 -0500 Subject: ML CPU pipeline: test py-torch-nvidia-apex (#33158) --- share/spack/gitlab/cloud_pipelines/stacks/ml-cpu/spack.yaml | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) (limited to 'share') 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 -- cgit v1.2.3-70-g09d2