From 3751f17264f6a341efe521fd421378fc153ec2fe Mon Sep 17 00:00:00 2001
From: Philipp Holl <philipp.holl@tum.de>
Date: Mon, 1 Jan 2024 19:45:37 +0100
Subject: [PATCH] Convert x0 in solve_linear()

---
 phiml/math/_optimize.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/phiml/math/_optimize.py b/phiml/math/_optimize.py
index 168067d3..5d4315b3 100644
--- a/phiml/math/_optimize.py
+++ b/phiml/math/_optimize.py
@@ -603,7 +603,8 @@ def solve_linear(f: Union[Callable[[X], Y], Tensor],
 
     if isinstance(f, Tensor) or (isinstance(f, LinearFunction) and prefer_explicit):  # Matrix solve
         if isinstance(f, LinearFunction):
-            matrix, bias = f.sparse_matrix_and_bias(solve.x0, *f_args, **f_kwargs)
+            x0 = math.convert(solve.x0)
+            matrix, bias = f.sparse_matrix_and_bias(x0, *f_args, **f_kwargs)
         else:
             matrix = f
             bias = 0