Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
tianlh
LightGBM-DCU
Commits
5c2b22c2
Commit
5c2b22c2
authored
Jun 07, 2017
by
Guolin Ke
Browse files
change socket set fail to warning.
parent
582ded51
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
14 additions
and
5 deletions
+14
-5
src/network/socket_wrapper.hpp
src/network/socket_wrapper.hpp
+13
-5
src/treelearner/gpu_tree_learner.h
src/treelearner/gpu_tree_learner.h
+1
-0
No files found.
src/network/socket_wrapper.hpp
View file @
5c2b22c2
...
...
@@ -81,8 +81,8 @@ inline int inet_pton(int af, const char *src, void *dst)
#define FREE(x) HeapFree(GetProcessHeap(), 0, (x))
namespace
SocketConfig
{
const
int
kSocketBufferSize
=
100
*
10
24
;
const
int
kMaxReceiveSize
=
100
*
10
24
;
const
int
kSocketBufferSize
=
100
*
10
00
;
const
int
kMaxReceiveSize
=
100
*
10
00
;
const
bool
kNoDelay
=
true
;
}
...
...
@@ -119,9 +119,17 @@ public:
if
(
sockfd_
==
INVALID_SOCKET
)
{
return
;
}
CHECK
(
setsockopt
(
sockfd_
,
SOL_SOCKET
,
SO_RCVBUF
,
reinterpret_cast
<
const
char
*>
(
&
SocketConfig
::
kSocketBufferSize
),
sizeof
(
SocketConfig
::
kSocketBufferSize
))
==
0
);
CHECK
(
setsockopt
(
sockfd_
,
SOL_SOCKET
,
SO_SNDBUF
,
reinterpret_cast
<
const
char
*>
(
&
SocketConfig
::
kSocketBufferSize
),
sizeof
(
SocketConfig
::
kSocketBufferSize
))
==
0
);
CHECK
(
setsockopt
(
sockfd_
,
IPPROTO_TCP
,
TCP_NODELAY
,
reinterpret_cast
<
const
char
*>
(
&
SocketConfig
::
kNoDelay
),
sizeof
(
SocketConfig
::
kNoDelay
))
==
0
);
if
(
setsockopt
(
sockfd_
,
SOL_SOCKET
,
SO_RCVBUF
,
reinterpret_cast
<
const
char
*>
(
&
SocketConfig
::
kSocketBufferSize
),
sizeof
(
SocketConfig
::
kSocketBufferSize
))
!=
0
)
{
Log
::
Warning
(
"Set SO_RCVBUF failed, please increase your net.core.rmem_max to 100k at least."
);
}
if
(
setsockopt
(
sockfd_
,
SOL_SOCKET
,
SO_SNDBUF
,
reinterpret_cast
<
const
char
*>
(
&
SocketConfig
::
kSocketBufferSize
),
sizeof
(
SocketConfig
::
kSocketBufferSize
))
!=
0
)
{
Log
::
Warning
(
"Set SO_SNDBUF failed, please increase your net.core.wmem_max to 100k at least."
);
}
if
(
setsockopt
(
sockfd_
,
IPPROTO_TCP
,
TCP_NODELAY
,
reinterpret_cast
<
const
char
*>
(
&
SocketConfig
::
kNoDelay
),
sizeof
(
SocketConfig
::
kNoDelay
))
!=
0
)
{
Log
::
Warning
(
"Set TCP_NODELAY failed."
);
}
}
inline
static
void
Startup
()
{
...
...
src/treelearner/gpu_tree_learner.h
View file @
5c2b22c2
...
...
@@ -267,6 +267,7 @@ namespace LightGBM {
class
GPUTreeLearner
:
public
SerialTreeLearner
{
public:
#pragma warning(disable : 4702)
explicit
GPUTreeLearner
(
const
TreeConfig
*
tree_config
)
:
SerialTreeLearner
(
tree_config
)
{
Log
::
Fatal
(
"GPU Tree Learner was not enabled in this build. Recompile with CMake option -DUSE_GPU=1"
);
}
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment