From dfe97dc82cba5463e552fed5550391dd8292b630 Mon Sep 17 00:00:00 2001 From: umiswing Date: Fri, 11 Aug 2023 09:09:51 +0800 Subject: [PATCH] Add flash attention to ci gpups. (#56117) --- tools/gpups_test.sh | 1 + 1 file changed, 1 insertion(+) diff --git a/tools/gpups_test.sh b/tools/gpups_test.sh index d0f9dd19341..fe2f93a34c9 100644 --- a/tools/gpups_test.sh +++ b/tools/gpups_test.sh @@ -46,6 +46,7 @@ parallel_list="^init_phi_test$|\ ^test_dist_fleet_ps11$|\ ^test_dist_fleet_ps12$|\ ^test_executor_feed_non_tensor$|\ +^test_flash_attention$|\ ^test_fused_adam_op$|\ ^test_fused_attention_no_dropout$|\ ^test_fused_attention_op$|\ -- GitLab