set(unittest_extra_libraries ActsPluginGnn)

add_unittest(GnnBoostTrackBuilding GnnBoostTrackBuildingTests.cpp)
add_unittest(GnnMetricHookTests GnnMetricHookTests.cpp)
if(ACTS_GNN_ENABLE_CUDA)
    add_unittest(ConnectedComponentsCuda ConnectedComponentCudaTests.cu)
    add_unittest(JunctionRemoval JunctionRemovalTests.cu)
endif()
add_unittest(Tensor TensorTests.cpp)
if(ACTS_GNN_ENABLE_TORCH)
    set(unittest_extra_libraries ActsPluginGnn ${TORCH_LIBRARIES})
    add_unittest(GnnTensorConversion GnnTensorConversionTests.cpp)
    add_unittest(GnnEdgeBuilding GnnEdgeBuildingTest.cpp)
endif()
