Skip to content

Conversation

@atalman
Copy link
Contributor

@atalman atalman commented Apr 28, 2025

Make sure with-cuda and with-cpu flags are correctly generating to aarch64 builds
Test:

python generate_binary_build_matrix.py --with-cpu disable  --with-cuda enable --with-rocm disable --operating-system linux-aarch64
{"include": [{"python_version": "3.9", "gpu_arch_type": "cuda-aarch64", "gpu_arch_version": "12.8-aarch64", "desired_cuda": "cu128", "container_image": "pytorch/manylinuxaarch64-builder:cuda12.8", "package_type": "wheel", "build_name": "wheel-py3_9-cuda-aarch6412_8-aarch64", "validation_runner": "linux.arm64.m7g.4xlarge", "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu128", "channel": "nightly", "upload_to_base_bucket": "no", "stable_version": "2.7.0", "use_split_build": false}]}

@vercel
Copy link

vercel bot commented Apr 28, 2025

@atalman is attempting to deploy a commit to the Meta Open Source Team on Vercel.

A member of the Team first needs to authorize it.

@facebook-github-bot facebook-github-bot added the CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. label Apr 28, 2025
@atalman atalman merged commit 971ac9d into pytorch:main Apr 28, 2025
44 of 45 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants