From 274b5a0d0c1634a26b03f00d8e9f2bf737d715c4 Mon Sep 17 00:00:00 2001 From: Connor Hack Date: Wed, 6 Nov 2024 10:16:19 -0800 Subject: [PATCH] Add WORLD_SIZE env var for torch.distributed --- .github/workflows/gha_workflow_llama_stack_tests.yml | 1 + 1 file changed, 1 insertion(+) diff --git a/.github/workflows/gha_workflow_llama_stack_tests.yml b/.github/workflows/gha_workflow_llama_stack_tests.yml index 402b04ef1..47b7f3c10 100644 --- a/.github/workflows/gha_workflow_llama_stack_tests.yml +++ b/.github/workflows/gha_workflow_llama_stack_tests.yml @@ -72,6 +72,7 @@ env: REPORTS_GEN: "" RANK: 0 + WORLD_SIZE: 4 jobs: execute_workflow: