Skip to content
This repository was archived by the owner on Aug 15, 2025. It is now read-only.

2nd attempt to guard aarch64-cuda (or rather cuda-aarch64)#1986

Merged
atalman merged 1 commit intopytorch:mainfrom
nWEIdia:main-fix-aarch64-cuda
Sep 12, 2024
Merged

2nd attempt to guard aarch64-cuda (or rather cuda-aarch64)#1986
atalman merged 1 commit intopytorch:mainfrom
nWEIdia:main-fix-aarch64-cuda

Conversation

@nWEIdia
Copy link
Copy Markdown
Contributor

@nWEIdia nWEIdia commented Sep 11, 2024

Previous PR did not seem to execute "pip3 install numpy -force-reinstall" and the CI did not seem to test this (only tested python 3.8).

cc @atalman @malfet @ptrblck @tinglvv

@tinglvv
Copy link
Copy Markdown
Contributor

tinglvv commented Sep 11, 2024

CUDA ARM wheel GPU_ARCH_TYPE was defined to be cuda-aarch64 to be in parallel with cpu aarch64 in https://github.com/pytorch/pytorch/pull/126174/files#diff-05a2f54752a171dcfec6fe4a27d17f307546ffb606d4e034414a3dd4f07192d0. Apologies for the confusion.

LGTM.

@atalman atalman merged commit 260aaf0 into pytorch:main Sep 12, 2024
@nWEIdia
Copy link
Copy Markdown
Contributor Author

nWEIdia commented Sep 12, 2024

https://github.com/pytorch/builder/actions/runs/10822278913/job/30025910519 looks good!

Linking "Previous PR" to be #1984 and issue to be #1985

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants