fixes gpu tests

Signed-off-by: Wenqi Li <wenqil@nvidia.com>

Author: wyli

monai/networks/nets/dints.py

@@ -153,8 +153,9 @@ def forward(self, x: torch.Tensor, weight: torch.Tensor):
             out: weighted average of the operation results.
         """
         out = 0.0
-        for idx, _ in enumerate(self.ops):
-            out = out + _(x) * weight[idx]
+        weight = weight.to(x)
+        for idx, _op in enumerate(self.ops):
+            out = out + _op(x) * weight[idx]
         return out