From 62f88df802680748dbf29ed621d1b9fcf4a4600e Mon Sep 17 00:00:00 2001 From: atalman Date: Mon, 29 Jan 2024 10:13:33 -0800 Subject: [PATCH 1/2] Modify nightly test --- .github/workflows/validate-nightly-pypi-wheel-binary-size.yml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/.github/workflows/validate-nightly-pypi-wheel-binary-size.yml b/.github/workflows/validate-nightly-pypi-wheel-binary-size.yml index a995ec817..82fb5e987 100644 --- a/.github/workflows/validate-nightly-pypi-wheel-binary-size.yml +++ b/.github/workflows/validate-nightly-pypi-wheel-binary-size.yml @@ -22,5 +22,5 @@ jobs: - name: Run validation run: | python tools/binary_size_validation/binary_size_validation.py \ - --url https://download.pytorch.org/whl/nightly/torch/ \ - --include "pypi" --only-latest-version --threshold 750 \ No newline at end of file + --url https://download.pytorch.org/whl/nightly/cu121/torch/ \ + --include "pypi" --only-latest-version --threshold 750 From 0b77fd605c517c38a98212319db299ce507973f3 Mon Sep 17 00:00:00 2001 From: atalman Date: Mon, 29 Jan 2024 10:27:18 -0800 Subject: [PATCH 2/2] fix --- .github/workflows/validate-nightly-pypi-wheel-binary-size.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/validate-nightly-pypi-wheel-binary-size.yml b/.github/workflows/validate-nightly-pypi-wheel-binary-size.yml index 82fb5e987..24fffc16e 100644 --- a/.github/workflows/validate-nightly-pypi-wheel-binary-size.yml +++ b/.github/workflows/validate-nightly-pypi-wheel-binary-size.yml @@ -23,4 +23,4 @@ jobs: run: | python tools/binary_size_validation/binary_size_validation.py \ --url https://download.pytorch.org/whl/nightly/cu121/torch/ \ - --include "pypi" --only-latest-version --threshold 750 + --include "linux" --only-latest-version --threshold 750