davrot/pytorch-sbs
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add files via upload

Browse source
This commit is contained in:
David Rotermund 2023-02-06 09:57:04 +01:00 committed by GitHub
parent 4a47c78c48
commit 6477fb29e2
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 73 additions and 59 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

22
network/h_dynamic_cnn_cpu_cpp/HDynamicCNNCPU.cpp
View file

@ -3,11 +3,12 @@
#include <omp.h>
#include <stdio.h>
#include <string.h>
#include <chrono>
#include <algorithm>
#include <cassert>
#include <iostream>
// #define DEBUGSHOWTIMEGLOBAL
HDynamicCNNCPU::HDynamicCNNCPU()
{
@ -113,6 +114,11 @@ void HDynamicCNNCPU::entrypoint(
assert((number_of_processes > 0));
omp_set_num_threads(number_of_processes);
#ifdef DEBUGSHOWTIMEGLOBAL
using TIME_resolution = std::chrono::nanoseconds;
auto TIME_start = std::chrono::high_resolution_clock::now();
#endif
#pragma omp parallel for
for (size_t pattern_id = 0; pattern_id < number_of_pattern; pattern_id++)
{
@ -141,6 +147,12 @@ void HDynamicCNNCPU::entrypoint(
pattern_id);
}
#ifdef DEBUGSHOWTIMEGLOBAL
auto TIME_end = std::chrono::high_resolution_clock::now();
float TIME_measured = TIME_resolution(TIME_end - TIME_start).count();
std::cout << "Time used : " << TIME_measured/(1000.0*1000.0) << "ms" << std::endl;
#endif
return;
};
@ -259,8 +271,10 @@ void HDynamicCNNCPU::update_one_ip(
spike = input_pointer + counter_spike * input_dim_c1;
if (*spike >= 0)
if (*spike < 0)
{
break;
}
if (epsilon_xy_dim_c0 != 0)
{
epsilon_subsegment =
@ -320,12 +334,12 @@ void HDynamicCNNCPU::update_one_ip(
}
else
{
epsilon_scale *= 1.0 + epsilon_subsegment * 1.0;
}
epsilon_scale *= 1.0 + epsilon_subsegment;
}
}
}
temp_value = 1.0 / epsilon_scale;
#pragma omp simd
for (size_t counter = 0; counter < h_dim; counter++)