| #include <cupti.h> |
| #include <stdio.h> |
| #include <cxxabi.h> |
| #include <time.h> |
|
|
| #define GPU_DEVICE 7 |
|
|
| #define BUF_SIZE (32 * 1024) |
| #define ALIGN_SIZE (8) |
| #define ALIGN_BUFFER(buffer, align) \ |
| (((uintptr_t) (buffer) & ((align)-1)) ? ((buffer) + (align) - ((uintptr_t) (buffer) & ((align)-1))) : (buffer)) |
|
|
| static uint64_t startTimestamp; |
| |
| |
|
|
| #define CUPTI_CALL(call) \ |
| do { \ |
| CUptiResult _status = call; \ |
| if (_status != CUPTI_SUCCESS) { \ |
| const char *errstr; \ |
| cuptiGetResultString(_status, &errstr); \ |
| fprintf(stderr, "%s:%d: error: function %s failed with error %s.\n", \ |
| __FILE__, __LINE__, #call, errstr); \ |
| if(_status == CUPTI_ERROR_LEGACY_PROFILER_NOT_SUPPORTED) \ |
| exit(0); \ |
| else \ |
| exit(-1); \ |
| } \ |
| } while (0) |
|
|
|
|
| extern inline __attribute__((always_inline)) unsigned long rdtsc() { |
| unsigned long a, d; |
|
|
| __asm__ volatile("rdtsc" : "=a"(a), "=d"(d)); |
|
|
| return (a | (d << 32)); |
| } |
|
|
| extern inline __attribute__((always_inline)) unsigned long rdtsp() { |
| struct timespec tms; |
| if (clock_gettime(CLOCK_REALTIME, &tms)) { |
| return -1; |
| } |
| unsigned long ns = tms.tv_sec * 1000000000; |
| ns += tms.tv_nsec; |
| return ns; |
| } |
|
|
| void initTrace(); |
| void finiTrace(); |
| void GPU_argv_init(); |
| void CUPTIAPI bufferRequested(uint8_t **buffer, size_t *size, size_t *maxNumRecords); |
| void CUPTIAPI bufferCompleted(CUcontext ctx, uint32_t streamId, uint8_t *buffer, size_t size, size_t validSize); |
| static void printActivity(CUpti_Activity *record); |
|
|
