Skip to content

Commit cc0e07b

Browse files
committed
modifying the error warning of missing libmpi libs
1 parent 15d681a commit cc0e07b

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

py/torch_tensorrt/dynamo/utils.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -977,9 +977,7 @@ def load_and_initialize_trtllm_plugin(plugin_lib_path: str) -> bool:
977977
except OSError as e_os_error:
978978
if "libmpi" in str(e_os_error):
979979
logger.warning(
980-
f"Failed to load libnvinfer_plugin_tensorrt_llm.so from {plugin_lib_path}. "
981-
f"The dependency libmpi.so is missing. "
982-
f"Please install the packages libmpich-dev and libopenmpi-dev.",
980+
f"Failed to load libnvinfer_plugin_tensorrt_llm.so from {plugin_lib_path}, got error {e_os_error} (hint: libmpi.so is a necessary dependency; ensure that OpenMPI or MPICH is installed on your system)",
983981
exc_info=e_os_error,
984982
)
985983
else:

0 commit comments

Comments
 (0)