//------------------------------------------------------------------------------
// GraphBLAS/CUDA/test/GB_cuda_timer.hpp
//------------------------------------------------------------------------------

// SuiteSparse:GraphBLAS, Timothy A. Davis, (c) 2017-2024, All Rights Reserved.
// This file: Copyright (c) 2024, NVIDIA CORPORATION. All rights reserved.
// SPDX-License-Identifier: Apache-2.0

//------------------------------------------------------------------------------

#ifndef GB_CUDA_TIMER_HPP
#define GB_CUDA_TIMER_HPP

#include <cuda_runtime.h>
struct GpuTimer
{
    cudaEvent_t start;
    cudaEvent_t stop;

    GpuTimer()
    {
        cudaEventCreate(&start);
        cudaEventCreate(&stop);
    }

    ~GpuTimer()
    {
        cudaEventDestroy(start);
        cudaEventDestroy(stop);
    }
         
    void Start()
    {
        cudaEventRecord(start, 0);
    }
               
    void Stop()
    {
        cudaEventRecord(stop, 0);
    }
                     
    float Elapsed()
    {
        float elapsed;
        cudaEventSynchronize(stop);
        cudaEventElapsedTime(&elapsed, start, stop);
        return elapsed;
    }
} ;

#endif