AI/rocm-systems
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #872 from yxsamliu/chrono

Browse Source 
Use chrono instead of sys/time.h in test
This commit is contained in:
Maneesh Gupta
2019-01-22 10:53:07 +05:30
committed by GitHub
parent e4a66e8471 aa6abb0dcf
commit 967a2d030d
1 changed files with 9 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/src/test_common.h
+9
View File
@@ -19,7 +19,11 @@ THE SOFTWARE.
#include <iostream> #include <iostream>
#include <iomanip> #include <iomanip>
#if __CUDACC__
#include <sys/time.h> #include <sys/time.h>
#else
#include <chrono>
#endif
#include <stddef.h> #include <stddef.h>
#include "hip/hip_runtime.h" #include "hip/hip_runtime.h"
@@ -105,9 +109,14 @@ namespace HipTest {
// Returns the current system time in microseconds // Returns the current system time in microseconds
inline long long get_time() { inline long long get_time() {
#if __CUDACC__
struct timeval tv; struct timeval tv;
gettimeofday(&tv, 0); gettimeofday(&tv, 0);
return (tv.tv_sec * 1000000) + tv.tv_usec; 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); double elapsed_time(long long startTimeUs, long long stopTimeUs);