diff --git a/.github/packaging/pre_build_gpu.sh b/.github/packaging/pre_build_gpu.sh index 9a7dd653b..267c8caca 100644 --- a/.github/packaging/pre_build_gpu.sh +++ b/.github/packaging/pre_build_gpu.sh @@ -49,6 +49,7 @@ build_monarch() { export CARGO_TERM_VERBOSE=true export CARGO_TERM_COLOR=always pip wheel --no-build-isolation --no-deps . -w "$WHL_DIR" + } append_date() { @@ -67,6 +68,15 @@ append_date() { fi } +smoke_test() { + cd "$WHL_DIR" + pip install --pre torch --no-cache-dir --index-url https://download.pytorch.org/whl/nightly/cu129 + pip install -r https://raw.githubusercontent.com/meta-pytorch/monarch/main/requirements.txt + pip install torchmonarch-0.0.1-cp310-cp310-linux_x86_64.whl + python -c "from monarch import actor" + echo "import successful" +} build_monarch -append_date \ No newline at end of file +append_date +smoke_test \ No newline at end of file diff --git a/.github/workflows/build_vllm.yaml b/.github/workflows/build_vllm.yaml index 0e8279ac4..4b6c59d2b 100644 --- a/.github/workflows/build_vllm.yaml +++ b/.github/workflows/build_vllm.yaml @@ -1,6 +1,7 @@ name: Build pinned vLLM against PyTorch nightly and upload on: + pull_request: push: branches: - nightly diff --git a/.github/workflows/build_wheels.yaml b/.github/workflows/build_wheels.yaml index a384991cd..40a07ab1e 100644 --- a/.github/workflows/build_wheels.yaml +++ b/.github/workflows/build_wheels.yaml @@ -1,6 +1,7 @@ name: Build nightly wheels and publish to PyTorch Index on: + pull_request: push: branches: - nightly