From f6b658e8b9cf66edfe9423c4d984cad2777216d0 Mon Sep 17 00:00:00 2001
From: Connor Hack <connorhack@meta.com>
Date: Wed, 6 Nov 2024 11:12:55 -0800
Subject: [PATCH] Add torchrun specially for meta_reference

---
 .github/workflows/gha_workflow_llama_stack_tests.yml | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/.github/workflows/gha_workflow_llama_stack_tests.yml b/.github/workflows/gha_workflow_llama_stack_tests.yml
index 47b7f3c10..2a7aa8b3d 100644
--- a/.github/workflows/gha_workflow_llama_stack_tests.yml
+++ b/.github/workflows/gha_workflow_llama_stack_tests.yml
@@ -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"