From 8323e6dae0b0f8f4b14ce6ee1689a03c5eadf180 Mon Sep 17 00:00:00 2001 From: Colman Glagovich Date: Tue, 17 Dec 2024 12:47:28 -0800 Subject: [PATCH] #0: Add chunked prefill test to frequent --- tests/scripts/t3000/run_t3000_frequent_tests.sh | 3 +++ 1 file changed, 3 insertions(+) diff --git a/tests/scripts/t3000/run_t3000_frequent_tests.sh b/tests/scripts/t3000/run_t3000_frequent_tests.sh index 3ade2f433552..fe58159bd239 100755 --- a/tests/scripts/t3000/run_t3000_frequent_tests.sh +++ b/tests/scripts/t3000/run_t3000_frequent_tests.sh @@ -67,6 +67,9 @@ run_t3000_llama3_tests() { echo "LOG_METAL: Llama3 tests for $llama_dir completed" done + # Run chunked prefill test for llama3-1B + LLAMA_DIR=$llama1b WH_ARCH_YAML=$wh_arch_yaml pytest models/demos/llama3/tests/test_llama_chunked_generation.py; fail+=$? + # Record the end time end_time=$(date +%s) duration=$((end_time - start_time))