diff --git a/projects/rocr-runtime/samples/common/hsatimer.h b/projects/rocr-runtime/samples/common/hsatimer.h index a15b06f583..2b7de9ed7e 100644 --- a/projects/rocr-runtime/samples/common/hsatimer.h +++ b/projects/rocr-runtime/samples/common/hsatimer.h @@ -2,21 +2,27 @@ #define __MYTIME__ // Will use AMD timer and general Linux timer based on users' need --> compilation flag - // need to consider platform is Windows or Linux #include #include #include -#include #include - #include #include #include using namespace std; -#include +#if defined(_MSC_VER) + #include + #include + #include +#else + #if defined(__GNUC__) + #include + #include + #endif // __GNUC__ +#endif //_MSC_VER #define HSA_FAILURE 1 #define HSA_SUCCESS 0