Skip to content

hessians/jacobians trigger scalar indexing for CuArrays #1348

Open
@marius311

Description

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.

Metadata

Assignees

No one assigned

    Labels

    CUDAAll things GPUsecond orderzygote over zygote, or otherwise

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions