Skip to content

Commit 566ae09

Browse files
test(lidar_centerpoint): added gtest for the cuda related parts of centerpoint (#7125)
* test: added gtest for the cuda related parts of centerpoint Signed-off-by: Kenzo Lobos-Tsunekawa <kenzo.lobos@tier4.jp> * chore: fixed ci/cd errors Signed-off-by: Kenzo Lobos-Tsunekawa <kenzo.lobos@tier4.jp> * style(pre-commit): autofix --------- Signed-off-by: Kenzo Lobos-Tsunekawa <kenzo.lobos@tier4.jp> Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
1 parent 3323c21 commit 566ae09

8 files changed

+1243
-0
lines changed

perception/lidar_centerpoint/CMakeLists.txt

+44
Original file line numberDiff line numberDiff line change
@@ -159,6 +159,50 @@ if(TRT_AVAIL AND CUDA_AVAIL AND CUDNN_AVAIL)
159159
ament_auto_add_gtest(test_nms
160160
test/test_nms.cpp
161161
)
162+
ament_auto_add_gtest(test_voxel_generator
163+
test/test_voxel_generator.cpp
164+
)
165+
166+
add_executable(test_preprocess_kernel
167+
test/test_preprocess_kernel.cpp
168+
lib/utils.cpp
169+
)
170+
171+
target_include_directories(test_preprocess_kernel PUBLIC
172+
${test_preprocess_kernel_SOURCE_DIR}
173+
)
174+
175+
target_link_libraries(test_preprocess_kernel
176+
centerpoint_cuda_lib
177+
gtest
178+
gtest_main
179+
)
180+
181+
ament_add_test(test_preprocess_kernel
182+
GENERATE_RESULT_FOR_RETURN_CODE_ZERO
183+
COMMAND "$<TARGET_FILE:test_preprocess_kernel>"
184+
)
185+
186+
add_executable(test_postprocess_kernel
187+
test/test_postprocess_kernel.cpp
188+
lib/utils.cpp
189+
)
190+
191+
target_include_directories(test_postprocess_kernel PUBLIC
192+
${test_postprocess_kernel_SOURCE_DIR}
193+
)
194+
195+
target_link_libraries(test_postprocess_kernel
196+
centerpoint_cuda_lib
197+
gtest
198+
gtest_main
199+
)
200+
201+
ament_add_test(test_postprocess_kernel
202+
GENERATE_RESULT_FOR_RETURN_CODE_ZERO
203+
COMMAND "$<TARGET_FILE:test_postprocess_kernel>"
204+
)
205+
162206
endif()
163207

164208
else()

perception/lidar_centerpoint/lib/preprocess/preprocess_kernel.cu

+5
Original file line numberDiff line numberDiff line change
@@ -78,6 +78,11 @@ cudaError_t generateVoxels_random_launch(
7878
{
7979
dim3 blocks((points_size + 256 - 1) / 256);
8080
dim3 threads(256);
81+
82+
if (blocks.x == 0) {
83+
return cudaGetLastError();
84+
}
85+
8186
generateVoxels_random_kernel<<<blocks, threads, 0, stream>>>(
8287
points, points_size, min_x_range, max_x_range, min_y_range, max_y_range, min_z_range,
8388
max_z_range, pillar_x_size, pillar_y_size, pillar_z_size, grid_y_size, grid_x_size, mask,

0 commit comments

Comments
 (0)