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