Skip to content

Commit ffe7557

Browse files
committed
Release FBGEMM v0.5.0
1 parent f8b3e30 commit ffe7557

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

.github/scripts/nova_dir.bash

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,11 @@ if [[ "$working_dir" == "$FBGEMM_REPO" ]]; then cd fbgemm_gpu || echo "Failed to
1313
## Build clean/wheel will be done in pre-script. Set flag such that setup.py will skip these steps in Nova workflow
1414
export BUILD_FROM_NOVA=1
1515

16+
## Override existing ENV VAR in Nova for Release
17+
export CHANNEL="test"
18+
export PIP_INSTALL_TORCH="pip install torch --index-url https://download.pytorch.org/whl/${CU_VERSION}"
19+
export PYTORCH_S3_BUCKET_PATH="s3://pytorch/whl/${CU_VERSION}/"
20+
1621
## Overwrite existing ENV VAR in Nova
1722
if [[ "$CONDA_ENV" != "" ]]; then export CONDA_RUN="conda run --no-capture-output -p ${CONDA_ENV}" && echo "$CONDA_RUN"; fi
1823
if [[ "$CU_VERSION" == "cu118" ]]; then export TORCH_CUDA_ARCH_LIST='7.0;8.0' && echo "$TORCH_CUDA_ARCH_LIST"; fi

0 commit comments

Comments
 (0)