fix gpu memory explosion problem when bthread is used

beb0e7ca · dongdaxiang · 9c58c6f9 · beb0e7ca
显示空白变更内容
内联并排

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