Linux & Windows 计时函数

直接上代码：

#if defined(_WIN32) && defined(_MSC_VER)
#include <windows.h>
double abtic() {
	__int64 freq;
	__int64 clock;
	QueryPerformanceFrequency( (LARGE_INTEGER *)&freq );
	QueryPerformanceCounter( (LARGE_INTEGER *)&clock );
	return (double)clock/freq*1000*1000;
}
#else
#include <time.h>
#include <sys/time.h>
double abtic() {
	double result = 0.0;
	struct timeval tv;
	gettimeofday( &tv, NULL );
	result = tv.tv_sec*1000*1000 + tv.tv_usec;
	return result;
}
#endif /* _WIN32 */

使用方法：

double timer = abtic();
...
printf("%fms\n", (abtic()-timer)*0.001);

可以用来调试的宏。

#if 1
  #define ABTMS(timer) timer=abtic();fprintf(stdout,"%4d  ",__LINE__)
  #define ABTME(timer) fprintf(stdout,"%4d  %8.8f ms\n",__LINE__,(abtic()-timer)*1.E3)
  #define SYCH cudaDeviceSynchronize()
#else
  #define ABTMS 
  #define ABTME 
  #define SYCH 
#endif