cuda_timer.h

Go to the documentation of this file.

/*
 *  SPDX-FileCopyrightText: Copyright 2021, Siavash Ameli <sameli@berkeley.edu>
 *  SPDX-License-Identifier: BSD-3-Clause
 *  SPDX-FileType: SOURCE
 *
 *  This program is free software: you can redistribute it and/or modify it
 *  under the terms of the license found in the LICENSE.txt file in the root
 *  directory of this source tree.
 */
 
 
#ifndef _CUDA_UTILITIES_CUDA_TIMER_H_
#define _CUDA_UTILITIES_CUDA_TIMER_H_
 
 
// =======
// Headers
// =======
 
#include <cuda_runtime_api.h>  // cudaEvent_t, cudaEventCreate,
                               // cudaEventDestropy, cudaEventRecord,
                               // cudaEventSynchronize, cudaEventElapsedTime
 
 
// ==========
// Cuda Timer
// ==========
 
 
class CudaTimer
{
    public:
 
        CudaTimer();
        ~CudaTimer();
        void start();
        void stop();
        float elapsed() const;
 
    protected:
        cudaEvent_t start_time;
        cudaEvent_t stop_time;
};
 
#endif  // _CUDA_UTILITIES_CUDA_TIMER_H_