Commit ddaa9530 by Neo Chien Committed by Tianqi Chen

[Contrib] Fix error message at callback_get_section_size() (#4221)

* [Contrib] Fix error message at callback_get_section_size()

* Trigger notification
parent 1eca1ad1
...@@ -22,6 +22,7 @@ from . import util ...@@ -22,6 +22,7 @@ from . import util
from .._ffi.base import py_str from .._ffi.base import py_str
from ..api import register_func from ..api import register_func
@register_func("tvm_callback_get_section_size") @register_func("tvm_callback_get_section_size")
def tvm_callback_get_section_size(binary_path, section_name, toolchain_prefix): def tvm_callback_get_section_size(binary_path, section_name, toolchain_prefix):
"""Finds size of the section in the binary. """Finds size of the section in the binary.
...@@ -53,7 +54,7 @@ def tvm_callback_get_section_size(binary_path, section_name, toolchain_prefix): ...@@ -53,7 +54,7 @@ def tvm_callback_get_section_size(binary_path, section_name, toolchain_prefix):
size_output = size_output.decode("utf-8") size_output = size_output.decode("utf-8")
if size_proc.returncode != 0: if size_proc.returncode != 0:
msg = "error in finding section size:\n" msg = "error in finding section size:\n"
msg += py_str(out) msg += py_str(size_output)
raise RuntimeError(msg) raise RuntimeError(msg)
# TODO(weberlo): Refactor this method and `*relocate_binary` so they are # TODO(weberlo): Refactor this method and `*relocate_binary` so they are
...@@ -93,6 +94,7 @@ def tvm_callback_get_section_size(binary_path, section_name, toolchain_prefix): ...@@ -93,6 +94,7 @@ def tvm_callback_get_section_size(binary_path, section_name, toolchain_prefix):
return section_size + 32 return section_size + 32
return section_size return section_size
@register_func("tvm_callback_relocate_binary") @register_func("tvm_callback_relocate_binary")
def tvm_callback_relocate_binary( def tvm_callback_relocate_binary(
binary_path, text_addr, rodata_addr, data_addr, bss_addr, toolchain_prefix): binary_path, text_addr, rodata_addr, data_addr, bss_addr, toolchain_prefix):
......
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