From b0853d69c835c725864826733d0d0d906ecd7934 Mon Sep 17 00:00:00 2001 From: giacomo m Date: Thu, 18 Apr 2024 23:04:30 +0200 Subject: [PATCH] Fix same test bug --- falkon/tests/test_conjgrad.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/falkon/tests/test_conjgrad.py b/falkon/tests/test_conjgrad.py index b4be47f..fc129cd 100644 --- a/falkon/tests/test_conjgrad.py +++ b/falkon/tests/test_conjgrad.py @@ -154,8 +154,8 @@ def test_restarts(self, data, centers, kernel, preconditioner, knm, kmm, vec_rhs def test_precomputed_kernel(self, data, centers, kernel, preconditioner, knm, kmm, vec_rhs, device): preconditioner = preconditioner.to(device) options = dataclasses.replace(self.basic_opt, use_cpu=device == "cpu") - knm = move_tensor(knm, device) - calc_kernel = PrecomputedKernel(knm, options) + knm_dev = move_tensor(knm, device) + calc_kernel = PrecomputedKernel(knm_dev, options) opt = FalkonConjugateGradient(calc_kernel, preconditioner, opt=options) # Solve (knm.T @ knm + lambda*n*kmm) x = knm.T @ b