diff --git a/spack.yaml b/spack.yaml
index 462c42f9e77e1e55bdb22bbf3705a79793ebbefe..13cb25ed029579ba82155b2c8d11565e0cc44072 100644
--- a/spack.yaml
+++ b/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