From ed408ac5995a4b34ae3a81b51e083bdadfdbba6d Mon Sep 17 00:00:00 2001 From: Nipun Gupta Date: Mon, 18 Aug 2025 07:30:50 -0700 Subject: [PATCH] Add --pre argument to fbgemm installer for OSS unit tests (#3288) Summary: Adding the --pre argument makes sure we can get the pre-release nightlies for fbgemm_gpu so we don't use outdated nightlies for tests Differential Revision: D80358634 --- .github/workflows/unittest_ci.yml | 2 +- .github/workflows/unittest_ci_cpu.yml | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/.github/workflows/unittest_ci.yml b/.github/workflows/unittest_ci.yml index 2f5d8a2dd..1c8cae063 100644 --- a/.github/workflows/unittest_ci.yml +++ b/.github/workflows/unittest_ci.yml @@ -82,7 +82,7 @@ jobs: conda run -n build_binary \ python -c "import torch.distributed" conda run -n build_binary \ - pip install fbgemm-gpu --index-url https://download.pytorch.org/whl/nightly/${{ matrix.cuda-tag }} + pip install --pre fbgemm-gpu --index-url https://download.pytorch.org/whl/nightly/${{ matrix.cuda-tag }} conda run -n build_binary \ python -c "import fbgemm_gpu" echo "fbgemm_gpu succeeded" diff --git a/.github/workflows/unittest_ci_cpu.yml b/.github/workflows/unittest_ci_cpu.yml index 7289fa509..518acd050 100644 --- a/.github/workflows/unittest_ci_cpu.yml +++ b/.github/workflows/unittest_ci_cpu.yml @@ -68,7 +68,7 @@ jobs: conda run -n build_binary \ python -c "import torch.distributed" conda run -n build_binary \ - pip install fbgemm-gpu --index-url https://download.pytorch.org/whl/nightly/cpu + pip install --pre fbgemm-gpu --index-url https://download.pytorch.org/whl/nightly/cpu conda run -n build_binary \ python -c "import fbgemm_gpu" echo "fbgemm_gpu succeeded"