diff --git a/examples/mixtral/pretrain_mixtral_8x7b_ptd.sh b/examples/mixtral/pretrain_mixtral_8x7b_ptd.sh index 5bc26dc78..26ef5f435 100644 --- a/examples/mixtral/pretrain_mixtral_8x7b_ptd.sh +++ b/examples/mixtral/pretrain_mixtral_8x7b_ptd.sh @@ -1,5 +1,5 @@ #!/bin/bash - +export ASCEND_LAUNCH_BLOCKING=1 export WITHOUT_JIT_COMPILE=1 export CUDA_DEVICE_MAX_CONNECTIONS=1