diff --git a/CMakeLists.txt b/CMakeLists.txt index 3362ededa..0e35abd8b 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -319,6 +319,8 @@ if(TRITON_BUILD_PYTHON_MODULE) LLVMPasses LLVMXPUCodeGen LLVMXPUAsmParser + LLVMXCNCodeGen + LLVMXCNAsmParser ) endif() diff --git a/README.md b/README.md index 306f775cf..066e64f6c 100644 --- a/README.md +++ b/README.md @@ -35,8 +35,9 @@ python3 -m pip install . --no-build-isolation -v ```shell # xpu (klx) # Recommended: Use the Docker image (22GB) https://su.bcebos.com/klx-sdk-release-public/xpytorch/docker/ubuntu2004_v030/ubuntu_2004_x86_64_v30.tar +# Contact kunlunxin-support@baidu.com for support mkdir -p ~/.flagtree/xpu; cd ~/.flagtree/xpu -wget https://github.com/FlagTree/flagtree/releases/download/v0.1.0-build-deps/XTDK-llvm19-ubuntu2004_x86_64.tar +wget https://github.com/FlagTree/flagtree/releases/download/v0.1.0-build-deps/XTDK-llvm19-ubuntu2004_x86_64.tar.gz wget https://github.com/FlagTree/flagtree/releases/download/v0.1.0-build-deps/xre-Linux-x86_64.tar.gz cd ${YOUR_CODE_DIR}/flagtree/python export FLAGTREE_BACKEND=xpu diff --git a/README_cn.md b/README_cn.md index 72d5e2acd..aa8c10403 100644 --- a/README_cn.md +++ b/README_cn.md @@ -35,8 +35,9 @@ python3 -m pip install . --no-build-isolation -v ```shell # xpu (klx) # 推荐使用镜像(22GB)https://su.bcebos.com/klx-sdk-release-public/xpytorch/docker/ubuntu2004_v030/ubuntu_2004_x86_64_v30.tar +# 联系 kunlunxin-support@baidu.com 可获取进一步支持 mkdir -p ~/.flagtree/xpu; cd ~/.flagtree/xpu -wget https://github.com/FlagTree/flagtree/releases/download/v0.1.0-build-deps/XTDK-llvm19-ubuntu2004_x86_64.tar +wget https://github.com/FlagTree/flagtree/releases/download/v0.1.0-build-deps/XTDK-llvm19-ubuntu2004_x86_64.tar.gz wget https://github.com/FlagTree/flagtree/releases/download/v0.1.0-build-deps/xre-Linux-x86_64.tar.gz cd ${YOUR_CODE_DIR}/flagtree/python export FLAGTREE_BACKEND=xpu diff --git a/python/setup_helper.py b/python/setup_helper.py index 12f134973..51780605a 100644 --- a/python/setup_helper.py +++ b/python/setup_helper.py @@ -338,7 +338,7 @@ def check_env(env_val): cache.store( file="XTDK-llvm19-ubuntu2004_x86_64", condition=("xpu" == flagtree_backend), - url="https://github.com/FlagTree/flagtree/releases/download/v0.1.0-build-deps/XTDK-llvm19-ubuntu2004_x86_64.tar", + url="https://github.com/FlagTree/flagtree/releases/download/v0.1.0-build-deps/XTDK-llvm19-ubuntu2004_x86_64.tar.gz", pre_hock=lambda: check_env('LLVM_BUILD_DIR'), post_hock=set_llvm_env, )