Add --pre argument to fbgemm installer for OSS unit tests

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

Author: nipung90

.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"

.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"