EngineX/xc-llm-kunlun
3
0
Fork 0
Code Issues Pull Requests Actions Projects Releases Wiki Activity

fix: use cuda visible (#244)

Browse Source 
Signed-off-by: lishaobing448 <shaobingli2024@163.com>
This commit is contained in:
lishaobing448
2026-03-02 17:33:13 +08:00
committed by GitHub
parent e4c9b9f988
commit dc63e81a7f
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
setup_env.sh
View File

@@ -1,5 +1,5 @@
unset XPU_DUMMY_EVENT unset XPU_DUMMY_EVENT
export XPU_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 export CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7
export XFT_USE_FAST_SWIGLU=1 #使用快速swiglu实现 export XFT_USE_FAST_SWIGLU=1 #使用快速swiglu实现
export XPU_USE_FAST_SWIGLU=1 #使用moe算子中快速swiglu实现 export XPU_USE_FAST_SWIGLU=1 #使用moe算子中快速swiglu实现
export XMLIR_CUDNN_ENABLED=1 export XMLIR_CUDNN_ENABLED=1