Update run_with_env_secrets.py

pytorch · Dec 5, 2024 · 4c8720d · 4c8720d
1 parent ad7114c
commit 4c8720d
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/.github/scripts/run_with_env_secrets.py b/.github/scripts/run_with_env_secrets.py
@@ -65,6 +65,7 @@ def main():
                 f"""
         docker run \
             -e PR_NUMBER \
+            -e HIP_VISIBLE_DEVICES=0 \
             -e RUNNER_ARTIFACT_DIR=/artifacts \
             -e RUNNER_DOCS_DIR=/docs \
             -e RUNNER_TEST_RESULTS_DIR=/test-results \