-
Notifications
You must be signed in to change notification settings - Fork 5.6k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
add mkldnn to paddle lib #10705
add mkldnn to paddle lib #10705
Conversation
… add-mkldnn-to-paddle-lib
cmake/inference_lib.cmake
Outdated
copy(mkldnn_lib | ||
SRCS ${MKLDNN_INSTALL_DIR} | ||
DSTS ${dst_dir} | ||
) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
目前打包效果如下:
mkldnn/
├── include
│ ├── mkldnn_debug.h
│ ├── mkldnn.h
│ ├── mkldnn.hpp
│ └── mkldnn_types.h
├── lib
│ ├── libmkldnn.so -> libmkldnn.so.0
│ ├── libmkldnn.so.0 -> libmkldnn.so.0.14.0
│ └── libmkldnn.so.0.14.0
├── libmkldnn.so.0
└── share
└── doc
└── mkldnn
└── LICENSE
可以精简下:
- 去掉share目录
- lib目录下只保留libmkldnn.so.0
mkldnn/
├── include
│ ├── mkldnn_debug.h
│ ├── mkldnn.h
│ ├── mkldnn.hpp
│ └── mkldnn_types.h
└── lib
└── libmkldnn.so.0
可以试下:
SRCS ${MKLDNN_INC_DIR} ${MKLDNN_SHARED_LIB}
DSTS ${dst} ${dst_dir}/lib
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
done
cmake/inference_lib.cmake
Outdated
set(dst_dir "${CMAKE_INSTALL_PREFIX}/third_party/install/mkldnn") | ||
copy(mkldnn_lib | ||
SRCS ${MKLDNN_INC_DIR} ${MKLDNN_SHARED_LIB} | ||
DSTS ${dst} ${dst_dir}/lib |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
105行第一个${dst}改成${dst_dir}
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
done
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM
… add-mkldnn-to-paddle-lib
task list: #10574