add gettid syscall for MAC OS

9 years ago · c7ece60e2d
parent 70cceb0cdd
commit c7ece60e2d
1 changed files with 8 additions and 4 deletions
--- a/paddle/utils/Stat.cpp
+++ b/paddle/utils/Stat.cpp
@ -23,10 +23,14 @@ namespace paddle {

 // return the thread id used by glog
 pid_t getTID() {
-#ifndef __NR_gettid
-#define __NR_gettid 224
-#endif
+  #if defined(__APPLE__) || defined(__OSX__)
+      pid_t tid = syscall(SYS_thread_selfid);
+  #elif defined(__LINUX__)
+      #ifndef __NR_gettid
+      #define __NR_gettid 224
+      #endif
      pid_t tid = syscall(__NR_gettid);
+  #endif
  CHECK_NE(tid, -1);
  return tid;
 }