From e87423cad4c64af4e4baa0772662c0ac0284cfc1 Mon Sep 17 00:00:00 2001
From: Stephen Nicholas Swatman <stephen.nicholas.swatman@cern.ch>
Date: Sat, 22 Jun 2024 12:56:01 +0200
Subject: [PATCH] Add SYCL tests for CCA code
Currently, we implement CCA tests only for CPU code and for CUDA code,
but we have an easily reusable test harness for this, so we might as
well test our SYCL code, too. This commit adds a new set of tests to the
SYCL suite which tests the functionality and accuracy of our CCL code.
---
tests/sycl/CMakeLists.txt | 1 +
tests/sycl/test_cca.sycl | 69 +++++++++++++++++++++++++++++++++++++++
2 files changed, 70 insertions(+)
create mode 100644 tests/sycl/test_cca.sycl
diff --git a/tests/sycl/CMakeLists.txt b/tests/sycl/CMakeLists.txt
index 0661870614..412532353b 100644
--- a/tests/sycl/CMakeLists.txt
+++ b/tests/sycl/CMakeLists.txt
@@ -18,6 +18,7 @@ traccc_add_test(
test_barrier.sycl
test_mutex.sycl
test_unique_lock.sycl
+ test_cca.sycl
LINK_LIBRARIES
GTest::gtest_main
diff --git a/tests/sycl/test_cca.sycl b/tests/sycl/test_cca.sycl
new file mode 100644
index 0000000000..3bb7de0b6a
--- /dev/null
+++ b/tests/sycl/test_cca.sycl
@@ -0,0 +1,69 @@
+/** TRACCC library, part of the ACTS project (R&D line)
+ *
+ * (c) 2022-2024 CERN for the benefit of the ACTS project
+ *
+ * Mozilla Public License Version 2.0
+ */
+
+#include <gtest/gtest.h>
+
+#include <CL/sycl.hpp>
+#include <functional>
+#include <vecmem/memory/host_memory_resource.hpp>
+#include <vecmem/memory/sycl/device_memory_resource.hpp>
+#include <vecmem/utils/sycl/async_copy.hpp>
+
+#include "tests/cca_test.hpp"
+#include "traccc/sycl/clusterization/clusterization_algorithm.hpp"
+
+namespace {
+
+cca_function_t f = [](const traccc::cell_collection_types::host& cells,
+ const traccc::cell_module_collection_types::host&
+ modules) {
+ std::map<traccc::geometry_id, vecmem::vector<traccc::measurement>> result;
+
+ vecmem::host_memory_resource host_mr;
+ cl::sycl::queue queue;
+ vecmem::sycl::device_memory_resource device_mr;
+ vecmem::sycl::async_copy copy{&queue};
+
+ traccc::sycl::clusterization_algorithm cc({device_mr}, copy, &queue, 1024);
+
+ traccc::cell_collection_types::buffer cells_buffer{
+ static_cast<traccc::cell_collection_types::buffer::size_type>(
+ cells.size()),
+ device_mr};
+ copy.setup(cells_buffer);
+ copy(vecmem::get_data(cells), cells_buffer)->ignore();
+
+ traccc::cell_module_collection_types::buffer modules_buffer{
+ static_cast<traccc::cell_module_collection_types::buffer::size_type>(
+ modules.size()),
+ device_mr};
+ copy.setup(modules_buffer);
+ copy(vecmem::get_data(modules), modules_buffer)->ignore();
+
+ auto measurements_buffer = cc(cells_buffer, modules_buffer);
+ traccc::measurement_collection_types::host measurements{&host_mr};
+ copy(measurements_buffer, measurements)->wait();
+
+ for (std::size_t i = 0; i < measurements.size(); i++) {
+ result[modules.at(measurements.at(i).module_link).surface_link.value()]
+ .push_back(measurements.at(i));
+ }
+
+ return result;
+};
+} // namespace
+
+TEST_P(ConnectedComponentAnalysisTests, Run) {
+ test_connected_component_analysis(GetParam());
+}
+
+INSTANTIATE_TEST_SUITE_P(
+ SYCLFastSvAlgorithm, ConnectedComponentAnalysisTests,
+ ::testing::Combine(
+ ::testing::Values(f),
+ ::testing::ValuesIn(ConnectedComponentAnalysisTests::get_test_files())),
+ ConnectedComponentAnalysisTests::get_test_name);