memoized_schedules.h

Go to the documentation of this file.

#ifndef FREE_TENSOR_MEMORIZED_SCHEDULES_H
#define FREE_TENSOR_MEMORIZED_SCHEDULES_H
 
#include <mutex>
#include <unordered_set>
 
#include <schedule/schedule_log.h>
 
namespace freetensor {
 
class MemoizedSchedules {
    std::unordered_set<ScheduleLog> memoized_;
    std::mutex lock_;
 
  public:
    ScheduleLog lookupOrCreate(const ScheduleLog &log) {
        std::lock_guard<std::mutex> guard(lock_);
        if (auto it = memoized_.find(log); it != memoized_.end()) {
            return *it;
        } else {
            memoized_.insert(log);
            return log;
        }
    }
};
 
} // namespace freetensor
 
#endif // FREE_TENSOR_MEMORIZED_SCHEDULES_H