mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-14 20:57:59 +00:00
fe9365f3f5
Migrated from pytorch/builder
Related to: https://github.com/pytorch/builder/issues/2054
Copying from : 3468139e81
Pull Request resolved: https://github.com/pytorch/pytorch/pull/143201
Approved by: https://github.com/seemethere, https://github.com/malfet
40 lines
1.4 KiB
YAML
40 lines
1.4 KiB
YAML
name: Test check_binary
|
|
|
|
on:
|
|
pull_request:
|
|
paths:
|
|
- .github/workflows/test-check-binary.yml
|
|
- .ci/pytorch/check_binary.sh
|
|
- .ci/pytorch//smoke_test/smoke_test.py
|
|
|
|
concurrency:
|
|
group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.sha }}-${{ github.event_name == 'workflow_dispatch' }}
|
|
cancel-in-progress: true
|
|
|
|
jobs:
|
|
check_binary_linux_cpu:
|
|
if: github.repository_owner == 'pytorch'
|
|
name: Test check_binary.sh for Linux CPU
|
|
uses: pytorch/test-infra/.github/workflows/linux_job.yml@main
|
|
with:
|
|
docker-image: python:3.11
|
|
docker-build-dir: "skip-docker-build"
|
|
script: |
|
|
pushd .ci/pytorch/
|
|
pip install --pre torch --index-url https://download.pytorch.org/whl/nightly/cpu
|
|
DESIRED_PYTHON=3.11 DESIRED_CUDA=cpu PACKAGE_TYPE=manywheel ./check_binary.sh
|
|
popd
|
|
|
|
check_binary_linux_cuda:
|
|
if: github.repository_owner == 'pytorch'
|
|
name: Test check_binary.sh for Linux CUDA
|
|
uses: pytorch/test-infra/.github/workflows/linux_job.yml@main
|
|
with:
|
|
runner: linux.4xlarge.nvidia.gpu
|
|
docker-image: python:3.11
|
|
docker-build-dir: "skip-docker-build"
|
|
script: |
|
|
pushd .ci/pytorch/
|
|
pip install --pre torch --index-url https://download.pytorch.org/whl/nightly/cu124
|
|
DESIRED_PYTHON=3.11 DESIRED_CUDA=cu124 PACKAGE_TYPE=manywheel ./check_binary.sh
|
|
popd
|