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.

Sign up for GitHub

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

Jump to bottom

hessians/jacobians trigger scalar indexing for CuArrays #1348

Open
marius311 opened this issue Jan 3, 2023 · 0 comments
Open

hessians/jacobians trigger scalar indexing for CuArrays #1348

marius311 opened this issue Jan 3, 2023 · 0 comments
Labels
CUDA All things GPU second order zygote over zygote, or otherwise

Comments

@marius311
Copy link
Contributor

Zygote v0.6.51 MWE:

using Zygote, CUDA
CUDA.allowscalar(false)
Zygote.hessian(x -> norm(sin.(x)), cu(rand(2)))

I think (but not tested) that this would be fixed by #1270, perhaps this can be added as a test there or wherever this is fixed.
@mcabbott mcabbott added CUDA All things GPU second order zygote over zygote, or otherwise labels Jan 3, 2023
@vpuri3 vpuri3 mentioned this issue Oct 26, 2023
Open
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
CUDA All things GPU second order zygote over zygote, or otherwise
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@marius311 @mcabbott