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

Labels

CUDA second order

Contributor

marius311 commented Jan 3, 2023

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 added CUDA second order labels

vpuri3 mentioned this issue

use ForwardDiff.jacobian in place of Zygote.forward_jacobian #1468

Open

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