Merge pull request #470 from guru4elephant/bthread_memory_problem

fix gpu memory explosion problem when bthread is used

Merge pull request #470 from guru4elephant/bthread_memory_problem
fix gpu memory explosion problem when bthread is used
9835553a · MRXLT · GitHub · 39bf2a1b · 6c381a3c · 9835553a
隐藏空白更改
内联并排

Showing with 4 addition and 0 deletion

core/predictor/common/macros.h core/predictor/common/macros.h +4 -0

未找到文件。
--- a/core/predictor/common/macros.h
+++ b/core/predictor/common/macros.h
@@ -27,6 +27,10 @@ namespace predictor {
  }
 #endif

+#ifdef WITH_GPU
+#define USE_PTHREAD
+#endif
+
 #ifdef USE_PTHREAD

 #define THREAD_T pthread_t