From 307b2ca33eda0f1da74d77a12891ef4ff69a3ee0 Mon Sep 17 00:00:00 2001 From: Connor Hack Date: Wed, 6 Nov 2024 09:58:35 -0800 Subject: [PATCH] Add RANK environment variable for torch.distributed --- .github/workflows/gha_workflow_llama_stack_tests.yml | 2 ++ 1 file changed, 2 insertions(+) diff --git a/.github/workflows/gha_workflow_llama_stack_tests.yml b/.github/workflows/gha_workflow_llama_stack_tests.yml index 514260528..402b04ef1 100644 --- a/.github/workflows/gha_workflow_llama_stack_tests.yml +++ b/.github/workflows/gha_workflow_llama_stack_tests.yml @@ -71,6 +71,8 @@ env: # Defines the output xml reports generated after a test is run REPORTS_GEN: "" + RANK: 0 + jobs: execute_workflow: name: Execute workload on Self-Hosted GPU k8s runner