Skip to content
Merged
2 changes: 1 addition & 1 deletion feectools
Submodule feectools updated 28 files
+112 −16 CUDA_STRATEGY.md
+3 −1 feectools/linalg/basic.py
+5 −2 feectools/linalg/block.py
+110 −22 feectools/linalg/direct_solvers.py
+3 −0 feectools/linalg/fft.py
+10 −7 feectools/linalg/kernels/stencil_dot_1d/stencil_dot_1d_cuda.cu
+15 −12 feectools/linalg/kernels/stencil_dot_1d/stencil_dot_1d_kernels.py
+12 −9 feectools/linalg/kernels/stencil_dot_2d/stencil_dot_2d_cuda.cu
+23 −75 feectools/linalg/kernels/stencil_dot_2d/stencil_dot_2d_kernels.py
+17 −29 feectools/linalg/kernels/stencil_dot_3d/stencil_dot_3d_cuda.cu
+33 −176 feectools/linalg/kernels/stencil_dot_3d/stencil_dot_3d_kernels.py
+12 −12 feectools/linalg/kernels/stencil_transpose_1d/stencil_transpose_1d_cuda.cu
+18 −17 feectools/linalg/kernels/stencil_transpose_1d/stencil_transpose_1d_kernels.py
+11 −12 feectools/linalg/kernels/stencil_transpose_2d/stencil_transpose_2d_cuda.cu
+35 −99 feectools/linalg/kernels/stencil_transpose_2d/stencil_transpose_2d_kernels.py
+32 −51 feectools/linalg/kernels/stencil_transpose_3d/stencil_transpose_3d_cuda.cu
+46 −229 feectools/linalg/kernels/stencil_transpose_3d/stencil_transpose_3d_kernels.py
+45 −0 feectools/linalg/kron.py
+62 −32 feectools/linalg/solvers.py
+48 −27 feectools/linalg/stencil.py
+149 −0 feectools/linalg/tests/inner_on_device_child.py
+34 −17 feectools/linalg/tests/kernel_test_args.py
+1 −1 feectools/linalg/tests/test_cuda_parity.py
+63 −20 feectools/linalg/tests/test_device_matvec.py
+160 −0 feectools/linalg/tests/test_inner_on_device.py
+284 −0 feectools/linalg/tests/test_kron_device_solve.py
+62 −0 feectools/linalg/tests/test_mpi_device.py
+1 −1 pyproject.toml
2 changes: 1 addition & 1 deletion pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -27,7 +27,7 @@ dependencies = [
"cunumpy >= 0.6.1, <= 0.6.1",
"pyccel>=2.2.0, <=2.2.3",
"maybempi>=0.1.2",
"feectools>=0.5.0, <=0.5.0",
"feectools>=0.6.0, <=0.6.0",
"scipy<=1.18.0",
"h5py<=3.16.0",
"h5netcdf<=1.8.1",
Expand Down
Loading