-
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
update xpu cmake for kunlun #33328
update xpu cmake for kunlun #33328
Conversation
Thanks for your contribution! |
cmake/external/xpu.cmake
Outdated
IF (HOST_SYSTEM MATCHES "ubuntu") | ||
SET(XPU_XRE_DIR_NAME "xre-ubuntu_x86_64") | ||
SET(XPU_XDNN_DIR_NAME "xdnn-ubuntu_x86_64") | ||
SET(XPU_XCCL_DIR_NAME "xccl-ubuntu_x86_64") |
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.
现在通讯库依赖厂内的brpc、protbuf等so,这些都是centos6u3编译的,所以这边只支持xccl-bdcentos_x86_64
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.
已更新
cmake/external/xpu.cmake
Outdated
SET(XPU_RT_LIB "${XPU_SDK_ROOT}/XTDK/runtime/shlib/libxpurt.so") | ||
SET(XPU_LIB_DIR "${XPU_SDK_ROOT}/XTDK/shlib/") | ||
endif() | ||
ELSE() |
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.
这边需要和萌萌和Lite的同学再对下,是否需要支持手动指定XPU_SDK_ROOT的方式,如果要支持这种方式,目录结构应该与自动下载是一样的。
cmake/external/xpu.cmake
Outdated
SET(XPU_XRE_DIR_NAME "xre-deepin_sw6_64") | ||
SET(XPU_XDNN_DIR_NAME "xdnn-deepin_sw6_64") | ||
SET(XPU_XCCL_DIR_NAME "xccl-deepin_sw6_64") | ||
ENDIF() |
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.
如果是bdcentos, cmake要怎么设置呢
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.
额 倒是没考虑到bdcentos,bdcentos应该不好区分吧,或者不同的系统版本在编译的时候指定,比如 WITH_BDCENTOS / WITH_UBUNTU
cmake/external/xpu.cmake
Outdated
SET(XPU_XCCL_DIR_NAME "xccl-deepin_sw6_64") | ||
ENDIF() | ||
|
||
SET(XPU_XRE_URL "https://baidu-kunlun-product.cdn.bcebos.com/KL-SDK/klsdk-dev/20210527/${XPU_XRE_DIR_NAME}.tar.gz" CACHE STRING "" FORCE) |
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.
这个base_url能独立出来吗,cmake的时候可以设定是不是会比较方便些
https://baidu-kunlun-product.cdn.bcebos.com/KL-SDK/klsdk-dev/20210527
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
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
PR types
Others
PR changes
Others
Describe
update xpu cmake for kunlun
*test=kunlun