FWCore/Utilities/test/RunningAverage_t.cpp - master - githubmirror/cmssw

FWCore/Utilities/test/RunningAverage_t.cpp
69 строк2 KB
Chris Jones
Use oneapi namespace
22 дек 2021, 00:32
22 дек 2021, 00:326cf8c1a
#include "FWCore/Utilities/interface/RunningAverage.h"

namespace {

  edm::RunningAverage localRA;

}

#include "oneapi/tbb/parallel_for.h"
#include "oneapi/tbb/task_arena.h"
#include "oneapi/tbb/global_control.h"
#include <iostream>
#include <vector>
#include <atomic>
#include <random>
#include <algorithm>
#include <type_traits>

namespace test_average {
  namespace running_average {
    int test() {
      oneapi::tbb::global_control control(
          oneapi::tbb::global_control::max_allowed_parallelism,
          oneapi::tbb::this_task_arena::max_concurrency());  // Explicit number of threads

      // std::random_device rd;
      std::mt19937 e2;  // (rd());
      std::normal_distribution<> normal_dist(1000., 200.);

      thread_local std::vector<float> v;

      int kk = 0;

      int n = 2000;

      int res[n];
      int qq[n];
      for (auto& y : qq)
        y = std::max(0., normal_dist(e2));

      auto theLoop = [&](int i) {
        kk++;
        v.reserve(res[i] = localRA.upper());
        v.resize(qq[i]);
        localRA.update(v.size());
        decltype(v) t;
        swap(v, t);
      };

      oneapi::tbb::parallel_for(oneapi::tbb::blocked_range<size_t>(0, n),
                                [&](const oneapi::tbb::blocked_range<size_t>& r) {
                                  for (size_t i = r.begin(); i < r.end(); ++i)
                                    theLoop(i);
                                });

      auto mm = std::max_element(res, res + n);
      std::cout << kk << ' ' << localRA.m_curr << ' ' << localRA.mean() << std::endl;
      for (auto& i : localRA.m_buffer)
        std::cout << i << ' ';
      std::cout << std::endl;
      std::cout << std::accumulate(res, res + n, 0) / n << ' ' << *std::min_element(res + 16, res + n) << ',' << *mm
                << std::endl;

      return 0;
    }
  }  // namespace running_average
}  // namespace test_average

int main() { return ::test_average::running_average::test(); }