From dc63e81a7fee6aad4d85d69b7adf0722200059f0 Mon Sep 17 00:00:00 2001 From: lishaobing448 Date: Mon, 2 Mar 2026 17:33:13 +0800 Subject: [PATCH] fix: use cuda visible (#244) Signed-off-by: lishaobing448 --- setup_env.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup_env.sh b/setup_env.sh index c282400..9b78322 100755 --- a/setup_env.sh +++ b/setup_env.sh @@ -1,5 +1,5 @@ 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 XPU_USE_FAST_SWIGLU=1 #使用moe算子中快速swiglu实现 export XMLIR_CUDNN_ENABLED=1