fix: bugfix device handling in thermo

Author: Oisin-M

src/earthkit/meteo/thermo/array/es_comp.py

@@ -148,7 +148,7 @@ def _es_mixed(t, xp):
     #   svp = alpha * es_water + (1.0 - alpha) * es_ice
 
     t = xp.asarray(t)
-    svp = xp.zeros(t.shape, dtype=t.dtype)
+    svp = xp.zeros(t.shape, dtype=t.dtype, device=xp.device(t))
 
     # ice range
     i_mask = t <= TI
@@ -176,7 +176,7 @@ def _es_ice_slope(t, xp):
 
 def _es_mixed_slope(t, xp):
     t = xp.asarray(t)
-    d_svp = xp.zeros(t.shape, dtype=t.dtype)
+    d_svp = xp.zeros(t.shape, dtype=t.dtype, device=xp.device(t))
 
     # ice range
     i_mask = t <= TI

src/earthkit/meteo/thermo/array/thermo.py

@@ -1058,7 +1058,7 @@ def compute_t_on_ma_stipanuk(self, ept, p):
         p = xp.asarray(p)
 
         size = xp.size(ept) if xp.size(ept) > xp.size(p) else xp.size(p)
-        t = xp.full(size, constants.T0 - 20, dtype=ept.dtype)
+        t = xp.full(size, constants.T0 - 20, dtype=ept.dtype, device=xp.device(ept))
 
         # if isinstance(p, np.ndarray):
         #     t = np.full(p.shape, constants.T0 - 20)
@@ -1083,7 +1083,7 @@ def compute_t_on_ma_davies(self, ept, p):
         ept = xp.asarray(ept)
         p = xp.asarray(p)
         if xp.size(ept) > xp.size(p):
-            p = xp.full(xp.size(ept), p, dtype=ept.dtype)
+            p = xp.full(xp.size(ept), p, dtype=ept.dtype, device=xp.device(ept))
 
         # p = xp.full(ept.shape, p, dtype=ept.dtype)