Fix py-torch build (maybe)

spack.yaml

@@ -33,5 +33,5 @@ spack:
     # demo for codejam12
     - funniest1022 %gcc@10.3.0
     # NRP
-    #- py-torch~mkldnn~cuda~cudnn~onnx_ml ^protobuf@:3.17.999 %gcc@10.3.0
+    - py-torch~mkldnn~cuda~cudnn~onnx_ml~rocm~tensorpipe~mpi~distributed ^protobuf@:3.17.999 %gcc@10.3.0
   concretization: together