Update environment_variables.sh for kunlunxin-cpm (#234)

FlagOpen · Sep 4, 2023 · 9cb8c63 · 9cb8c63
1 parent 1390dda
commit 9cb8c63
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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