ML CPU pipeline: test py-torch-nvidia-apex (#33158)

2022-10-10 13:18:05 -05:00 · 2022-10-10 13:18:05 -05:00 · 831d7979ca
commit 831d7979ca
parent 14f6de9bf2
1 changed files with 1 additions and 2 deletions
--- 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