InducingPointKernel eye now correctly inherits dtype
@@ -43,7 +43,7 @@ class InducingPointKernel(Kernel):
return self._cached_kernel_inv_root
else:
chol = psd_safe_cholesky(self._inducing_mat, upper=True)
- eye = torch.eye(chol.size(-1), device=chol.device)
+ eye = torch.eye(chol.size(-1), device=chol.device, dtype=chol.dtype)
inv_root = torch.triangular_solve(eye, chol)[0]
res = inv_root
