fix

huggingface · Jan 12, 2024 · b4d875e · b4d875e
1 parent 83df91c
commit b4d875e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/.github/workflows/slow-tests.yml b/.github/workflows/slow-tests.yml
@@ -6,7 +6,6 @@ on:
 env:
   RUN_SLOW: "yes"
   IS_GITHUB_CI: "1"
-  CUDA_LAUNCH_BLOCKING: "1"
   SLACK_API_TOKEN: ${{ secrets.SLACK_CIFEEDBACK_BOT_TOKEN }}
 
 

diff --git a/Makefile b/Makefile
@@ -29,13 +29,13 @@ slow_dpo_tests:
 run_sft_examples:
 	touch temp_results_sft_tests.txt
 	for file in $(ACCELERATE_CONFIG_PATH)/*.yaml; do \
-		TRL_ACCELERATE_CONFIG=$${file} bash $(COMMAND_FILES_PATH)/run_sft.sh; \
+		CUDA_LAUNCH_BLOCKING=1 TRL_ACCELERATE_CONFIG=$${file} bash $(COMMAND_FILES_PATH)/run_sft.sh; \
 		echo $$?','$${file} >> temp_results_sft_tests.txt; \
 	done
 
 run_dpo_examples:
 	touch temp_results_dpo_tests.txt
 	for file in $(ACCELERATE_CONFIG_PATH)/*.yaml; do \
-		TRL_ACCELERATE_CONFIG=$${file} bash $(COMMAND_FILES_PATH)/run_dpo.sh; \
+		CUDA_LAUNCH_BLOCKING=1 TRL_ACCELERATE_CONFIG=$${file} bash $(COMMAND_FILES_PATH)/run_dpo.sh; \
 		echo $$?','$${file} >> temp_results_dpo_tests.txt; \
 	done