Fixed typo in transpose_2d_kernel and matvec_2d_kernel arguments

Author: max-models

psydac/linalg/stencil_dot_kernels.py

@@ -26,7 +26,7 @@ def matvec_1d_kernel(mat: 'float[:, :]',
     out[p_out + i1_loc] = val
 
 
-def matvec_2d_kernel(mat: 'float[:, :, :, :, :]',
+def matvec_2d_kernel(mat: 'float[:, :, :, :]',
                      x: 'float[:, :]',
                      out: 'float[:, :]',
                      s_in: 'int[:]',
@@ -112,7 +112,8 @@ def matvec_2d_kernel(mat: 'float[:, :, :, :, :]',
             val += mat[p_out[0] + i1_loc,
                        p_out[1] + i2_loc,
                        d1, d2] * x[i1 + d1 - s_in[0],
-                                   i2 + d2 - s_in[1]]
+                                    i2 + d2 - s_in[1]]
+
     out[p_out[0] + i1_loc,
         p_out[1] + i2_loc] = val
 

psydac/linalg/stencil_transpose_kernels.py

@@ -24,8 +24,8 @@ def transpose_1d_kernel(mat: 'float[:, :]',
 
         matT[p_out + i1_loc, d1] = mat[p_in + j1_loc, p_out + i1 - j1]
 
-def transpose_2d_kernel(mat: 'float[:, :, :, :, :]',
-                        matT: 'float[:, :, :, :, :]',
+def transpose_2d_kernel(mat: 'float[:, :, :, :]',
+                        matT: 'float[:, :, :, :]',
                         s_in: 'int[:]',  # refers to matT
                         p_in: 'int[:]',
                         add: 'int[:]',