mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
cbc53b7696
Upgrade to PyBind11 v3. This allows us to strip out our own (possibly broken?) handling of the C++ ABI when building extensions, in favor of the more-complete PyBind11 internal handling. Fixes a few test failures due to https://github.com/pybind/pybind11/issues/5774, which effectively makes the `__qualname__` attribute of functions platform-dependent. Test plan: CI Pull Request resolved: https://github.com/pytorch/pytorch/pull/160754 Approved by: https://github.com/Skylion007
43 lines
1.6 KiB
YAML
43 lines
1.6 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_v2.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_v2.yml@main
|
|
with:
|
|
runner: linux.g4dn.4xlarge.nvidia.gpu
|
|
docker-image: python:3.11
|
|
docker-build-dir: "skip-docker-build"
|
|
script: |
|
|
STABLE_CUDA_VERSION=$(python3 .github/scripts/get_ci_variable.py --cuda-stable-version)
|
|
CUDA_VERSION_NODOT=$(echo ${STABLE_CUDA_VERSION} | tr -d '.')
|
|
pushd .ci/pytorch/
|
|
pip install --pre torch --index-url https://download.pytorch.org/whl/nightly/cu${CUDA_VERSION_NODOT}
|
|
DESIRED_PYTHON=3.11 DESIRED_CUDA=cu${CUDA_VERSION_NODOT} PACKAGE_TYPE=manywheel ./check_binary.sh
|
|
popd
|