Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "623ae79500c12d04fdafc315acbcf6edcef5aac3"
Unverified Commit 60a4a13e authored by James Lamb's avatar James Lamb Committed by GitHub
Browse files

[python-package] remove unnecessary allocations in ctypes code (#6111)

parent 7c9a985a
Hide whitespace changes
Inline Side-by-side
Showing with 8 additions and 8 deletions
python-package/lightgbm/basic.py
View file @ 60a4a13e
...@@ -479,7 +479,7 @@ class _ConfigAliases: ...@@ -479,7 +479,7 @@ class _ConfigAliases:
buffer_len = 1 << 20 buffer_len = 1 << 20
tmp_out_len = ctypes.c_int64(0) tmp_out_len = ctypes.c_int64(0)
string_buffer = ctypes.create_string_buffer(buffer_len) string_buffer = ctypes.create_string_buffer(buffer_len)
ptr_string_buffer = ctypes.c_char_p(*[ctypes.addressof(string_buffer)]) ptr_string_buffer = ctypes.c_char_p(ctypes.addressof(string_buffer))
_safe_call(_LIB.LGBM_DumpParamAliases( _safe_call(_LIB.LGBM_DumpParamAliases(
ctypes.c_int64(buffer_len), ctypes.c_int64(buffer_len),
ctypes.byref(tmp_out_len), ctypes.byref(tmp_out_len),
...@@ -488,7 +488,7 @@ class _ConfigAliases: ...@@ -488,7 +488,7 @@ class _ConfigAliases:
# if buffer length is not long enough, re-allocate a buffer # if buffer length is not long enough, re-allocate a buffer
if actual_len > buffer_len: if actual_len > buffer_len:
string_buffer = ctypes.create_string_buffer(actual_len) string_buffer = ctypes.create_string_buffer(actual_len)
ptr_string_buffer = ctypes.c_char_p(*[ctypes.addressof(string_buffer)]) ptr_string_buffer = ctypes.c_char_p(ctypes.addressof(string_buffer))
_safe_call(_LIB.LGBM_DumpParamAliases( _safe_call(_LIB.LGBM_DumpParamAliases(
ctypes.c_int64(actual_len), ctypes.c_int64(actual_len),
ctypes.byref(tmp_out_len), ctypes.byref(tmp_out_len),
...@@ -3315,7 +3315,7 @@ class Booster: ...@@ -3315,7 +3315,7 @@ class Booster:
buffer_len = 1 << 20 buffer_len = 1 << 20
tmp_out_len = ctypes.c_int64(0) tmp_out_len = ctypes.c_int64(0)
string_buffer = ctypes.create_string_buffer(buffer_len) string_buffer = ctypes.create_string_buffer(buffer_len)
ptr_string_buffer = ctypes.c_char_p(*[ctypes.addressof(string_buffer)]) ptr_string_buffer = ctypes.c_char_p(ctypes.addressof(string_buffer))
_safe_call(_LIB.LGBM_BoosterGetLoadedParam( _safe_call(_LIB.LGBM_BoosterGetLoadedParam(
self._handle, self._handle,
ctypes.c_int64(buffer_len), ctypes.c_int64(buffer_len),
...@@ -3325,7 +3325,7 @@ class Booster: ...@@ -3325,7 +3325,7 @@ class Booster:
# if buffer length is not long enough, re-allocate a buffer # if buffer length is not long enough, re-allocate a buffer
if actual_len > buffer_len: if actual_len > buffer_len:
string_buffer = ctypes.create_string_buffer(actual_len) string_buffer = ctypes.create_string_buffer(actual_len)
ptr_string_buffer = ctypes.c_char_p(*[ctypes.addressof(string_buffer)]) ptr_string_buffer = ctypes.c_char_p(ctypes.addressof(string_buffer))
_safe_call(_LIB.LGBM_BoosterGetLoadedParam( _safe_call(_LIB.LGBM_BoosterGetLoadedParam(
self._handle, self._handle,
ctypes.c_int64(actual_len), ctypes.c_int64(actual_len),
...@@ -4078,7 +4078,7 @@ class Booster: ...@@ -4078,7 +4078,7 @@ class Booster:
buffer_len = 1 << 20 buffer_len = 1 << 20
tmp_out_len = ctypes.c_int64(0) tmp_out_len = ctypes.c_int64(0)
string_buffer = ctypes.create_string_buffer(buffer_len) string_buffer = ctypes.create_string_buffer(buffer_len)
ptr_string_buffer = ctypes.c_char_p(*[ctypes.addressof(string_buffer)]) ptr_string_buffer = ctypes.c_char_p(ctypes.addressof(string_buffer))
_safe_call(_LIB.LGBM_BoosterSaveModelToString( _safe_call(_LIB.LGBM_BoosterSaveModelToString(
self._handle, self._handle,
ctypes.c_int(start_iteration), ctypes.c_int(start_iteration),
...@@ -4091,7 +4091,7 @@ class Booster: ...@@ -4091,7 +4091,7 @@ class Booster:
# if buffer length is not long enough, re-allocate a buffer # if buffer length is not long enough, re-allocate a buffer
if actual_len > buffer_len: if actual_len > buffer_len:
string_buffer = ctypes.create_string_buffer(actual_len) string_buffer = ctypes.create_string_buffer(actual_len)
ptr_string_buffer = ctypes.c_char_p(*[ctypes.addressof(string_buffer)]) ptr_string_buffer = ctypes.c_char_p(ctypes.addressof(string_buffer))
_safe_call(_LIB.LGBM_BoosterSaveModelToString( _safe_call(_LIB.LGBM_BoosterSaveModelToString(
self._handle, self._handle,
ctypes.c_int(start_iteration), ctypes.c_int(start_iteration),
...@@ -4146,7 +4146,7 @@ class Booster: ...@@ -4146,7 +4146,7 @@ class Booster:
buffer_len = 1 << 20 buffer_len = 1 << 20
tmp_out_len = ctypes.c_int64(0) tmp_out_len = ctypes.c_int64(0)
string_buffer = ctypes.create_string_buffer(buffer_len) string_buffer = ctypes.create_string_buffer(buffer_len)
ptr_string_buffer = ctypes.c_char_p(*[ctypes.addressof(string_buffer)]) ptr_string_buffer = ctypes.c_char_p(ctypes.addressof(string_buffer))
_safe_call(_LIB.LGBM_BoosterDumpModel( _safe_call(_LIB.LGBM_BoosterDumpModel(
self._handle, self._handle,
ctypes.c_int(start_iteration), ctypes.c_int(start_iteration),
...@@ -4159,7 +4159,7 @@ class Booster: ...@@ -4159,7 +4159,7 @@ class Booster:
# if buffer length is not long enough, reallocate a buffer # if buffer length is not long enough, reallocate a buffer
if actual_len > buffer_len: if actual_len > buffer_len:
string_buffer = ctypes.create_string_buffer(actual_len) string_buffer = ctypes.create_string_buffer(actual_len)
ptr_string_buffer = ctypes.c_char_p(*[ctypes.addressof(string_buffer)]) ptr_string_buffer = ctypes.c_char_p(ctypes.addressof(string_buffer))
_safe_call(_LIB.LGBM_BoosterDumpModel( _safe_call(_LIB.LGBM_BoosterDumpModel(
self._handle, self._handle,
ctypes.c_int(start_iteration), ctypes.c_int(start_iteration),
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment