diff --git a/tests/src/test_common.h b/tests/src/test_common.h index da5b47ea47..bacbf35a22 100644 --- a/tests/src/test_common.h +++ b/tests/src/test_common.h @@ -19,7 +19,11 @@ THE SOFTWARE. #include #include +#if __CUDACC__ +#include +#else #include +#endif #include #include "hip/hip_runtime.h" @@ -105,8 +109,14 @@ namespace HipTest { // Returns the current system time in microseconds inline long long get_time() { +#if __CUDACC__ + struct timeval tv; + gettimeofday(&tv, 0); + return (tv.tv_sec * 1000000) + tv.tv_usec; +#else return std::chrono::high_resolution_clock::now().time_since_epoch() /std::chrono::microseconds(1); +#endif } double elapsed_time(long long startTimeUs, long long stopTimeUs);