diff --git a/python/setup.py.in b/python/setup.py.in index 7a357762508bf2ac51eadfb0e5039343c324058d..5506443733650631fe045be3f701a41519352e8d 100644 --- a/python/setup.py.in +++ b/python/setup.py.in @@ -130,7 +130,8 @@ if '${WITH_MKLDNN}' == 'ON': # The reason is that all thirdparty libraries in the same directory, # thus, libmkldnn.so.0 will find libmklml_intel.so and libiomp5.so. command = "patchelf --set-rpath '$ORIGIN/' ${MKLDNN_SHARED_LIB}" - os.system(command) + if os.system(command) != 0: + raise Exception("patchelf --set-rpath for libmkldnn.so.0 fails") package_data['paddle.libs']+=['libmkldnn.so.0'] shutil.copy('${MKLDNN_SHARED_LIB}', libs_path) # remove unused paddle/libs/__init__.py @@ -142,7 +143,8 @@ package_dir['paddle.libs']=libs_path # core.so is in paddle.fluid, thus paddle/fluid/../libs will pointer to above libraries. # This operation will fix https://github.com/PaddlePaddle/Paddle/issues/3213 command = "patchelf --set-rpath '$ORIGIN/../libs/' ${PADDLE_BINARY_DIR}/python/paddle/fluid/core.so" -os.system(command) +if os.system(command) != 0: + raise Exception("patchelf --set-rpath for core.so fails") setup(name='${PACKAGE_NAME}', version='${PADDLE_VERSION}',