[BUILD] [DOC] [BACKEND] fix xpu download link and build support (#2)

zhzhcookie · web-flow · commit 337857054d67 · 2025-05-23T17:14:12.000+08:00
diff --git 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
@@ -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
@@ -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
@@ -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,
 )

Original file line number	Diff line number	Diff line change
`@@ -319,6 +319,8 @@ if(TRITON_BUILD_PYTHON_MODULE)`
`319`	`319`	`LLVMPasses`
`320`	`320`	`LLVMXPUCodeGen`
`321`	`321`	`LLVMXPUAsmParser`
	`322`	`+ LLVMXCNCodeGen`
	`323`	`+ LLVMXCNAsmParser`
`322`	`324`	`)`
`323`	`325`	`endif()`
`324`	`326`
Original file line number	Diff line number	Diff line change
`@@ -338,7 +338,7 @@ def check_env(env_val):`
`338`	`338`	`cache.store(`
`339`	`339`	`file="XTDK-llvm19-ubuntu2004_x86_64",`
`340`	`340`	`condition=("xpu" == flagtree_backend),`
`341`		`- url="https://github.com/FlagTree/flagtree/releases/download/v0.1.0-build-deps/XTDK-llvm19-ubuntu2004_x86_64.tar",`
	`341`	`+ url="https://github.com/FlagTree/flagtree/releases/download/v0.1.0-build-deps/XTDK-llvm19-ubuntu2004_x86_64.tar.gz",`
`342`	`342`	`pre_hock=lambda: check_env('LLVM_BUILD_DIR'),`
`343`	`343`	`post_hock=set_llvm_env,`
`344`	`344`	`)`