timer.hpp

Go to the documentation of this file.

// SPDX-FileCopyrightText: 2025 Povl Filip Sonne-Frederiksen
//
// SPDX-License-Identifier: GPL-3.0-or-later
 
#pragma once
#include "reusex/core/logging.hpp"
#include "reusex/vision/tensor_rt/common/check.hpp"
 
namespace reusex::vision::tensor_rt::nv {
 
class EventTimer {
    public:
  EventTimer() {
    checkRuntime(cudaEventCreate(&begin_));
    checkRuntime(cudaEventCreate(&end_));
  }
 
  virtual ~EventTimer() {
    checkRuntime(cudaEventDestroy(begin_));
    checkRuntime(cudaEventDestroy(end_));
  }
 
  void start(cudaStream_t stream = nullptr) {
    stream_ = stream;
    checkRuntime(cudaEventRecord(begin_, stream));
  }
 
  float stop(const char *prefix = "timer") {
    float times = 0;
    checkRuntime(cudaEventRecord(end_, stream_));
    checkRuntime(cudaEventSynchronize(end_));
    checkRuntime(cudaEventElapsedTime(&times, begin_, end_));
    reusex::info("[⏰ {}] : {:.5f} ms", prefix, times);
    return times;
  }
 
    private:
  cudaStream_t stream_ = nullptr;
  cudaEvent_t begin_ = nullptr, end_ = nullptr;
};
 
}; // namespace reusex::vision::tensor_rt::nv