diff --git a/gloo/transport/tcp/device.cc b/gloo/transport/tcp/device.cc index e56c2ad37..d89d24d97 100644 --- a/gloo/transport/tcp/device.cc +++ b/gloo/transport/tcp/device.cc @@ -7,7 +7,7 @@ */ #include "gloo/transport/tcp/device.h" - +#include #include #include #include diff --git a/gloo/types.h b/gloo/types.h index 32d50a2be..6aa26d672 100644 --- a/gloo/types.h +++ b/gloo/types.h @@ -1,5 +1,6 @@ #pragma once +#include #include #ifdef __CUDA_ARCH__