8f43927575
examples/ is based on KasperskyLab/knp at 55470ed6 with local changes to the mnist-learn altai/blifat models, cuda-backend CMake and model_loader. Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
103 lines
4 KiB
C++
103 lines
4 KiB
C++
/**
|
|
* @file synchronization_test.cpp
|
|
* @brief Synchronization tests for different backends.
|
|
* @kaspersky_support Vartenkov An.
|
|
* @date 17.05.24.
|
|
* @license Apache 2.0
|
|
* @copyright © 2024 AO Kaspersky Lab
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
|
|
#include <knp/backends/cpu-multi-threaded/backend.h>
|
|
#include <knp/backends/cpu-single-threaded/backend.h>
|
|
#include <knp/core/population.h>
|
|
#include <knp/framework/synchronization.h>
|
|
|
|
#include <generators.h>
|
|
#include <spdlog/spdlog.h>
|
|
#include <tests_common.h>
|
|
|
|
#include <vector>
|
|
|
|
|
|
namespace knp::testing
|
|
{
|
|
class STestingBack : public knp::backends::single_threaded_cpu::SingleThreadedCPUBackend
|
|
{
|
|
public:
|
|
STestingBack() = default;
|
|
void _init() override { knp::backends::single_threaded_cpu::SingleThreadedCPUBackend::_init(); }
|
|
};
|
|
|
|
|
|
class MTestingBack : public knp::backends::multi_threaded_cpu::MultiThreadedCPUBackend
|
|
{
|
|
public:
|
|
MTestingBack() = default;
|
|
void _init() override { knp::backends::multi_threaded_cpu::MultiThreadedCPUBackend::_init(); }
|
|
};
|
|
|
|
} // namespace knp::testing
|
|
|
|
|
|
TEST(SynchronizationSuite, SingleThreadCpuTest)
|
|
{
|
|
// Create a single-neuron neural network: input -> input_projection -> population <=> loop_projection.
|
|
knp::testing::STestingBack backend;
|
|
knp::testing::BLIFATPopulation population{knp::testing::neuron_generator, 1};
|
|
auto loop_projection =
|
|
knp::testing::DeltaProjection{population.get_uid(), population.get_uid(), knp::testing::synapse_generator, 1};
|
|
auto input_projection = knp::testing::DeltaProjection{
|
|
knp::core::UID{false}, population.get_uid(), knp::testing::input_projection_gen, 1};
|
|
|
|
backend.load_populations({population});
|
|
backend.load_projections({input_projection, loop_projection});
|
|
backend._init();
|
|
|
|
auto network = knp::framework::synchronization::get_network_copy(backend);
|
|
ASSERT_EQ(network.get_projections().size(), 2);
|
|
ASSERT_EQ(network.get_populations().size(), 1);
|
|
auto &proj0 = std::get<knp::testing::DeltaProjection>(network.get_projections()[0]);
|
|
auto &proj1 = std::get<knp::testing::DeltaProjection>(network.get_projections()[1]);
|
|
auto &pop = std::get<knp::testing::BLIFATPopulation>(network.get_populations()[0]);
|
|
ASSERT_EQ(proj0.size(), 1);
|
|
ASSERT_EQ(proj1.size(), 1);
|
|
ASSERT_EQ(pop.size(), 1);
|
|
}
|
|
|
|
|
|
TEST(SynchronizationSuite, MultiThreadCpuTest)
|
|
{
|
|
// Create a single-neuron neural network: input -> input_projection -> population <=> loop_projection.
|
|
knp::testing::MTestingBack backend;
|
|
knp::testing::BLIFATPopulation population{knp::testing::neuron_generator, 1};
|
|
auto loop_projection =
|
|
knp::testing::DeltaProjection{population.get_uid(), population.get_uid(), knp::testing::synapse_generator, 1};
|
|
auto input_projection = knp::testing::DeltaProjection{
|
|
knp::core::UID{false}, population.get_uid(), knp::testing::input_projection_gen, 1};
|
|
|
|
backend.load_populations({population});
|
|
backend.load_projections({input_projection, loop_projection});
|
|
backend._init();
|
|
|
|
auto network = knp::framework::synchronization::get_network_copy(backend);
|
|
ASSERT_EQ(network.get_projections().size(), 2);
|
|
ASSERT_EQ(network.get_populations().size(), 1);
|
|
auto &proj0 = std::get<knp::testing::DeltaProjection>(network.get_projections()[0]);
|
|
auto &proj1 = std::get<knp::testing::DeltaProjection>(network.get_projections()[1]);
|
|
auto &pop = std::get<knp::testing::BLIFATPopulation>(network.get_populations()[0]);
|
|
ASSERT_EQ(proj0.size(), 1);
|
|
ASSERT_EQ(proj1.size(), 1);
|
|
ASSERT_EQ(pop.size(), 1);
|
|
}
|