phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-07-30 07:39:38 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add torchrun specially for meta_reference

Browse source
This commit is contained in:
Connor Hack 2024-11-06 11:12:55 -08:00
parent 274b5a0d0c
commit f6b658e8b9
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
.github/workflows/gha_workflow_llama_stack_tests.yml vendored
View file

@ -239,8 +239,8 @@ jobs:
for file in "$dir"/test_*.py; do
test_name=$(basename "$file")
new_file="result-${dir_name}-${test_name}.xml"
if pytest -s -v ${TESTS_PATH}/${dir_name}/${test_name} -m "meta_reference and llama_3b" \
--junitxml="${{ github.workspace }}/${new_file}"; then
if torchrun $(which pytest) -s -v ${TESTS_PATH}/${dir_name}/${test_name} -m "meta_reference and llama_3b" \
--junitxml="${{ github.workspace }}/${new_file}"; then
echo "Test passed: $test_name"
else
echo "Test failed: $test_name"