fix option names

pytorch · Mar 20, 2024 · 01e1359 · 01e1359
1 parent b34384b
commit 01e1359
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/.github/workflows/test_build_wheels_linux_without_cuda.yml b/.github/workflows/test_build_wheels_linux_without_cuda.yml
@@ -42,7 +42,7 @@ jobs:
       ref: nightly
       test-infra-repository: ${{ github.repository }}
       test-infra-ref: ${{ github.ref }}
-      build-matrix: ${{ needs.generate-matrix.outputs.matrix }}
+      build-matrix: ${{ needs.generate-matrix-default.outputs.matrix }}
       pre-script: ${{ matrix.pre-script }}
       post-script: ${{ matrix.post-script }}
       smoke-test-script: ${{ matrix.smoke-test-script }}
@@ -66,7 +66,7 @@ jobs:
       ref: nightly
       test-infra-repository: ${{ github.repository }}
       test-infra-ref: ${{ github.ref }}
-      build-matrix: ${{ needs.generate-matrix.outputs.matrix }}
+      build-matrix: ${{ needs.generate-matrix-github-hosted.outputs.matrix }}
       pre-script: packaging/install_torchdata.sh
       post-script: ""
       smoke-test-script: test/smoke_tests/smoke_tests.py

diff --git a/tools/scripts/generate_binary_build_matrix.py b/tools/scripts/generate_binary_build_matrix.py
@@ -650,7 +650,7 @@ def main(args) -> None:
         default=os.getenv("USE_ONLY_DL_PYTORCH_ORG", "false"),
     )
     parser.add_argument(
-        "--use-github-runners",
+        "--use-github-hosted-runners",
         help="Use GitHub Hosted Runners?",
         type=str,
         choices=[ENABLE, DISABLE],