Hugging Face's logo

Spaces:
USTC3DVer
/
Lite3DReg
Sleeping

App Files Community
Lite3DReg / cpp /tools /OmpHelper.h
duanbotu123
Initial commit: add index.html
f6dd1c2
Raw History Blame Contribute Delete
1.51 kB
#ifndef OMPHELPER_H_
#define OMPHELPER_H_
#define USE_OPENMP
#ifdef USE_OPENMP
#include <omp.h>
#ifdef USE_MSVC
#define OMP_PARALLEL __pragma(omp parallel)
#define OMP_FOR __pragma(omp for)
#define OMP_SINGLE __pragma(omp single)
#define OMP_SECTIONS __pragma(omp sections)
#define OMP_SECTION __pragma(omp section)
#else
#define OMP_PARALLEL _Pragma("omp parallel")
#define OMP_FOR _Pragma("omp for")
#define OMP_SINGLE _Pragma("omp single")
#define OMP_SECTIONS _Pragma("omp sections")
#define OMP_SECTION _Pragma("omp section")
#endif
#else
#include <ctime>
#define OMP_PARALLEL
#define OMP_FOR
#define OMP_SINGLE
#define OMP_SECTIONS
#define OMP_SECTION
#endif
#include <cassert>
#include <vector>
class Timer
{
public:
typedef int EventID;
EventID get_time()
{
EventID id = time_values_.size();
#ifdef USE_OPENMP
time_values_.push_back(omp_get_wtime());
#else
time_values_.push_back(clock());
#endif
return id;
}
double elapsed_time(EventID event1, EventID event2)
{
assert(event1 >= 0 && event1 < static_cast<EventID>(time_values_.size()));
assert(event2 >= 0 && event2 < static_cast<EventID>(time_values_.size()));
#ifdef USE_OPENMP
return time_values_[event2] - time_values_[event1];
#else
return double(time_values_[event2] - time_values_[event1]) / CLOCKS_PER_SEC;
#endif
}
void reset()
{
time_values_.clear();
}
private:
#ifdef USE_OPENMP
std::vector<double> time_values_;
#else
std::vector<clock_t> time_values_;
#endif
};
#endif /* OMPHELPER_H_ */