diff --git a/python/rainbow/cuda/geometry/grid3.py b/python/rainbow/cuda/geometry/grid3.py index a8395ea..cc5ff99 100644 --- a/python/rainbow/cuda/geometry/grid3.py +++ b/python/rainbow/cuda/geometry/grid3.py @@ -149,7 +149,6 @@ def get_gradient_device(p: float64[:], min_coord: float64[:], spacing: float64[: def is_inside_device(p: float64[:], min_coord: float64[:], max_coord: float64[:], boundary: float64) -> boolean: """ This function test if the given spatial 3D point is inside the given grid and the boundary value. - Args: p (float64[:]): The 3D spatial point. min_coord (float64[:]): The minimum corner of the grid. diff --git a/python/rainbow/cuda/unit_tests/test_math_matrix_kernel.py b/python/rainbow/cuda/unit_tests/test_math_matrix_kernel.py index c423876..1fdfdbc 100644 --- a/python/rainbow/cuda/unit_tests/test_math_matrix_kernel.py +++ b/python/rainbow/cuda/unit_tests/test_math_matrix_kernel.py @@ -87,6 +87,7 @@ def mat33_dot_vec3_kernel(M: float64[:, :, :], V3: float64[:, :], result: float6 tid = cuda.threadIdx.x Matrix.mat33_dot_vec3(M[tid, :, :], V3[tid, :], result[tid, :]) + @cuda.jit def mat43_T_kernel(M: float64[:, :, :], result: float64[:, :, :]): """ Unit test kernel function for mat43_T_device diff --git a/python/unit_tests/test_cuda_geometry_grid3.py b/python/unit_tests/test_cuda_geometry_grid3.py index 5c55ceb..af2a334 100644 --- a/python/unit_tests/test_cuda_geometry_grid3.py +++ b/python/unit_tests/test_cuda_geometry_grid3.py @@ -11,11 +11,11 @@ import rainbow.cuda.unit_tests.test_geometry_grid3_kernel as TEST_GRID3K - def simpelfunc(coord): _, _, z = coord[0], coord[1], coord[2] return z + @unittest.skipIf(not cuda.is_available(), "CUDA not available") class TestGrid3Cuda(unittest.TestCase):