From 9cb8c632327db4cef68c8632c9a0071210f09c4a Mon Sep 17 00:00:00 2001
From: KungYork <30741085+KungYork@users.noreply.github.com>
Date: Mon, 4 Sep 2023 11:04:58 +0800
Subject: [PATCH] Update environment_variables.sh for kunlunxin-cpm (#234)

---
 training/kunlunxin/cpm-pytorch/config/environment_variables.sh | 1 +
 1 file changed, 1 insertion(+)

diff --git a/training/kunlunxin/cpm-pytorch/config/environment_variables.sh b/training/kunlunxin/cpm-pytorch/config/environment_variables.sh
index 6bac80921..e3977276b 100644
--- a/training/kunlunxin/cpm-pytorch/config/environment_variables.sh
+++ b/training/kunlunxin/cpm-pytorch/config/environment_variables.sh
@@ -16,6 +16,7 @@ touch topo_file
 export XPUSIM_TOPOLOGY_FILE=$(readlink -f $topo_file)
 
 ## workaround due to ccix bug
+export BKCL_PCIE_RING=1
 export ALLREDUCE_ASYNC="0"
 export ALLREDUCE_FUSION="0"
 export BKCL_FORCE_SYNC=1