forked from lijiext/lammps
108 lines
3.2 KiB
C++
108 lines
3.2 KiB
C++
/* -----------------------------------------------------------------------
|
|
Copyright (2010) Sandia Corporation. Under the terms of Contract
|
|
DE-AC04-94AL85000 with Sandia Corporation, the U.S. Government retains
|
|
certain rights in this software. This software is distributed under
|
|
the Simplified BSD License.
|
|
----------------------------------------------------------------------- */
|
|
|
|
#ifndef HIP_TIMER_H
|
|
#define HIP_TIMER_H
|
|
|
|
|
|
#include <hip/hip_runtime.h>
|
|
#include "hip_macros.h"
|
|
#include "hip_device.h"
|
|
|
|
namespace ucl_hip {
|
|
|
|
/// Class for timing CUDA Driver events
|
|
class UCL_Timer {
|
|
public:
|
|
inline UCL_Timer() : _total_time(0.0f), _initialized(false) { }
|
|
inline UCL_Timer(UCL_Device &dev) : _total_time(0.0f), _initialized(false)
|
|
{ init(dev); }
|
|
|
|
inline ~UCL_Timer() { clear(); }
|
|
|
|
/// Clear any data associated with timer
|
|
/** \note init() must be called to reuse timer after a clear() **/
|
|
inline void clear() {
|
|
if (_initialized) {
|
|
CU_DESTRUCT_CALL(hipEventDestroy(start_event));
|
|
CU_DESTRUCT_CALL(hipEventDestroy(stop_event));
|
|
_initialized=false;
|
|
_total_time=0.0;
|
|
}
|
|
}
|
|
|
|
/// Initialize default command queue for timing
|
|
inline void init(UCL_Device &dev) { init(dev, dev.cq()); }
|
|
|
|
/// Initialize command queue for timing
|
|
inline void init(UCL_Device &dev, command_queue &cq) {
|
|
clear();
|
|
_cq=cq;
|
|
_initialized=true;
|
|
CU_SAFE_CALL( hipEventCreateWithFlags(&start_event,0) );
|
|
CU_SAFE_CALL( hipEventCreateWithFlags(&stop_event,0) );
|
|
}
|
|
|
|
/// Start timing on command queue
|
|
inline void start() { CU_SAFE_CALL(hipEventRecord(start_event,_cq)); }
|
|
|
|
/// Stop timing on command queue
|
|
inline void stop() { CU_SAFE_CALL(hipEventRecord(stop_event,_cq)); }
|
|
|
|
/// Block until the start event has been reached on device
|
|
inline void sync_start()
|
|
{ CU_SAFE_CALL(hipEventSynchronize(start_event)); }
|
|
|
|
/// Block until the stop event has been reached on device
|
|
inline void sync_stop()
|
|
{ CU_SAFE_CALL(hipEventSynchronize(stop_event)); }
|
|
|
|
/// Set the time elapsed to zero (not the total_time)
|
|
inline void zero() {
|
|
CU_SAFE_CALL(hipEventRecord(start_event,_cq));
|
|
CU_SAFE_CALL(hipEventRecord(stop_event,_cq));
|
|
}
|
|
|
|
/// Set the total time to zero
|
|
inline void zero_total() { _total_time=0.0; }
|
|
|
|
/// Add time from previous start and stop to total
|
|
/** Forces synchronization **/
|
|
inline double add_to_total()
|
|
{ double t=time(); _total_time+=t; return t/1000.0; }
|
|
|
|
/// Add a user specified time to the total (ms)
|
|
inline void add_time_to_total(const double t) { _total_time+=t; }
|
|
|
|
/// Return the time (ms) of last start to stop - Forces synchronization
|
|
inline double time() {
|
|
float timer;
|
|
CU_SAFE_CALL(hipEventSynchronize(stop_event));
|
|
CU_SAFE_CALL( hipEventElapsedTime(&timer,start_event,stop_event) );
|
|
return timer;
|
|
}
|
|
|
|
/// Return the time (s) of last start to stop - Forces synchronization
|
|
inline double seconds() { return time()/1000.0; }
|
|
|
|
/// Return the total time in ms
|
|
inline double total_time() { return _total_time; }
|
|
|
|
/// Return the total time in seconds
|
|
inline double total_seconds() { return _total_time/1000.0; }
|
|
|
|
private:
|
|
hipEvent_t start_event, stop_event;
|
|
hipStream_t _cq;
|
|
double _total_time;
|
|
bool _initialized;
|
|
};
|
|
|
|
} // namespace
|
|
|
|
#endif
|