From df737041e4a9f3f55978cc74db9a9cea27fa9fa0 Mon Sep 17 00:00:00 2001
From: Determinant <ted.sybil@gmail.com>
Date: Fri, 5 Jun 2015 10:58:57 +0800
Subject: add profiling; add ce accurarcy; several other changes

---
 matrix/cuda_helper.h | 26 ++++++++++++++++++++++++--
 1 file changed, 24 insertions(+), 2 deletions(-)

(limited to 'matrix/cuda_helper.h')

diff --git a/matrix/cuda_helper.h b/matrix/cuda_helper.h
index cedc643..5e5f2ad 100644
--- a/matrix/cuda_helper.h
+++ b/matrix/cuda_helper.h
@@ -1,17 +1,23 @@
 #ifndef NERV_CUDA_HELPER_H
 #define NERV_CUDA_HELPER_H
+#include "cuda.h"
+#include "cuda_runtime.h"
+#include "driver_types.h"
+#include "cublas_v2.h"
 #define CUBLAS_SAFE_CALL(call) \
     do { \
         cublasStatus_t  err = (call); \
         if (err != CUBLAS_STATUS_SUCCESS) \
-            nerv_error(L, "cumatrix cublas error: %s", cublasGetErrorString(err)); \
+            nerv_error(L, "cumatrix cublas error: %s at %s:%d", \
+                        cublasGetErrorString(err), __FILE__, __LINE__); \
     } while (0)
 
 #define CUDA_SAFE_CALL(call) \
     do { \
         cudaError_t err = (call); \
         if (err != cudaSuccess) \
-            nerv_error(L, "cumatrix CUDA error: %s", cudaGetErrorString(err)); \
+            nerv_error(L, "cumatrix CUDA error: %s at %s:%d", \
+                            cudaGetErrorString(err), __FILE__, __LINE__); \
     } while (0)
 
 #define CUDA_SAFE_SYNC_CALL(call) \
@@ -52,4 +58,20 @@ static const char *cublasGetErrorString(cublasStatus_t err) {
     }
     return "<unknown>";
 }
+
+#define PROFILE_START \
+    do { \
+        cudaEvent_t start, stop; \
+        cudaEventCreate(&start); \
+        cudaEventCreate(&stop); \
+        cudaEventRecord(start, 0);
+#define PROFILE_STOP \
+        cudaEventRecord(stop, 0); \
+        cudaEventSynchronize(stop); \
+        float milliseconds = 0; \
+        cudaEventElapsedTime(&milliseconds, start, stop); \
+        accu_profile(__func__, milliseconds / 1000); \
+    } while (0);
+
+#define PROFILE_END 
 #endif
-- 
cgit v1.2.3