Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Allow single indexing of arrays of GPU arrays #760

Merged
merged 2 commits into from
Dec 14, 2023

Conversation

ToucheSir
Copy link
Contributor

This fixes FluxML/Zygote.jl#1470, which was caused by Zygote switching to the CR rule. I can't think of any scenarios where this wouldn't work and the current implementation would, so hopefully it's a non-controversial fix.

@ToucheSir
Copy link
Contributor Author

Test failures all look to be unrelated to this PR. Does anyone know what's going on there?

@oxinabox
Copy link
Member

the CI / Julia 1.6 - ubuntu-latest - x86 has been broken for ages.
It is I assume a julia bug since it is a segfault.
and I don't track Nightly failures.

@oxinabox oxinabox merged commit ae37562 into JuliaDiff:main Dec 14, 2023
8 of 11 checks passed
@ToucheSir ToucheSir deleted the bc/getindex-gpu-aoa branch December 14, 2023 14:45
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

Assignment to multiple arrays is not differentiable on GPU since Zygote.jl 0.6.67
2 participants