misc: minor update for debugging & compilation (dmlc#92)

eric-haibin-lin · Aug 4, 2021 · f25af22 · f25af22
1 parent 80c53bc
commit f25af22
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/src/ucx_van.h b/src/ucx_van.h
@@ -989,7 +989,8 @@ class UCXVan : public Van {
           }
         }
         if (trial >= max_retry) {
-          CHECK(false) << "ucp_listener_create failed: " << ucs_status_string(status);
+          CHECK(false) << "ucp_listener_create failed: " << ucs_status_string(status)
+                       << " node=" << node.DebugString();
         }
       }
     }

diff --git a/src/van_common.h b/src/van_common.h
@@ -17,6 +17,7 @@
 #define PS_RDMA_COMMON_H_
 
 #if defined(DMLC_USE_RDMA) || defined(DMLC_USE_FABRIC)
+#include <unistd.h>
 
 #define DIVUP(x, y) (((x) + (y)-1) / (y))
 #define ROUNDUP(x, y) (DIVUP((x), (y)) * (y))