Removed default precompiled=False in set_backend. Added precompiled=True where needed in test_stencil_matrix.py

Author: max-models

psydac/linalg/stencil.py

@@ -945,8 +945,10 @@ def __init__( self, V, W, pads=None , backend=None, precompiled=False):
         self._transpose_args = self._prepare_transpose_args()
         self._transpose_func = kernels['transpose'][self._ndim]
 
-        # if backend is None:
-        #     backend = PSYDAC_BACKENDS.get(os.environ.get('PSYDAC_BACKEND')) or PSYDAC_BACKENDS['python']
+        if backend is None:
+            backend = PSYDAC_BACKENDS.get(os.environ.get('PSYDAC_BACKEND')) or PSYDAC_BACKENDS['python']
+        print(backend)
+        print(precompiled)
         self.set_backend(backend, precompiled)
 
     #--------------------------------------
@@ -1764,7 +1766,7 @@ def _prepare_transpose_args(self):
         return args
 
     # ...
-    def set_backend(self, backend, precompiled=False):
+    def set_backend(self, backend, precompiled):
         '''
         Define which kernels are called when using .dot() and .transpose()
         
@@ -1774,7 +1776,7 @@ def set_backend(self, backend, precompiled=False):
             Psydac backend option.
             
         precompiled : bool
-            Whether to use precompiled kernels. Default: False.
+            Whether to use precompiled kernels.
         '''
         from psydac.api.ast.linalg import LinearOperatorDot
         self._backend = backend

psydac/linalg/tests/test_stencil_matrix.py

@@ -59,14 +59,14 @@ def test_stencil_matrix_1d_serial_init(dtype, n1, p1, s1, P1=True):
 
     # Create a vector space V and a matrix M from V to V
     V = StencilVectorSpace(cart, dtype=dtype)
-    M = StencilMatrix(V, V)
+    M = StencilMatrix(V, V, precompiled=True)
 
     # Check properties of this matrix
     assert M.domain == V
     assert M.codomain == V
     assert M.dtype == dtype
     assert M.pads == (p1,)
-    assert M.backend == None
+    assert M.backend == PSYDAC_BACKEND_PYTHON
     assert M._data.shape == (n1 + 2 * p1 * s1, 1 + 2 * p1)
     assert M.shape == (n1 , n1 )
 
@@ -89,14 +89,14 @@ def test_stencil_matrix_2d_serial_init(dtype, n1, n2, p1, p2, s1, s2, P1=True, P
 
     # Create a vector space V and a matrix M from V to V
     V = StencilVectorSpace(cart, dtype=dtype)
-    M = StencilMatrix(V, V)
+    M = StencilMatrix(V, V, precompiled=True)
 
     # Check properties of this matrix
     assert M.domain == V
     assert M.codomain == V
     assert M.dtype == dtype
     assert M.pads == (p1, p2)
-    assert M.backend == None
+    assert M.backend == PSYDAC_BACKEND_PYTHON
     assert M._data.shape == (n1 + 2 * p1 * s1, n2 + 2 * p2 * s2, 1 + 2 * p1, 1 + 2 * p2)
     assert M.shape == (n1 * n2, n1 * n2)
 
@@ -129,7 +129,7 @@ def test_stencil_matrix_3d_serial_init(dtype, n1, n2, n3, p1, p2, p3, s1, s2, s3
     assert M.codomain == V
     assert M.dtype == dtype
     assert M.pads == (p1, p2, p3)
-    assert M.backend == None
+    assert M.backend == PSYDAC_BACKEND_PYTHON
     assert M._data.shape == (n1 + 2 * p1 * s1, n2 + 2 * p2 * s2, n3+2*p3*s3, 1 + 2 * p1, 1 + 2 * p2, 1+2*p3)
     assert M.shape == (n1 * n2 * n3, n1 * n2 * n3)
 
@@ -302,7 +302,7 @@ def test_stencil_matrix_2d_math(dtype, n1, n2, p1, p2, s1, s2, P1=True, P2=False
         assert m.codomain == V
         assert m.dtype == dtype
         assert m.pads == (p1, p2)
-        assert m.backend == None
+        assert M.backend == PSYDAC_BACKEND_PYTHON
         assert m._data.shape == (n1 + 2 * p1 * s1, n2 + 2 * p2 * s2, 1 + 2 * p1, 1 + 2 * p2)
         assert m.shape == (n1 * n2, n1 * n2)
         assert np.allclose(m._data, m_exa)