fixing device mismatch in botorch optim stopping

Summary: This fixes a device mismatch problem in botorch optimization stopping procedure. It makes sure that "weights " and "fvals" are tensors with the same device type.

Reviewed By: Balandat

Differential Revision: D20480298

fbshipit-source-id: 6932de24afc8e6a3a67ae0175f34990fd8e1d0d2

Author: Abbas Kazerouni

botorch/optim/stopping.py

@@ -107,6 +107,7 @@ def evaluate(self, fvals: Tensor) -> bool:
             return False
 
         weights = self.weights
+        weights = weights.to(fvals)
         if self._prev_fvals.ndim > 1:
             weights = weights.unsqueeze(-1)