Custom `rrule` not working with ReverseDiff

Abierto
  • ¿Algo no está bien?
#208 1 comentario 1 reacción 0 asignados Ver en GitHub

Nadie ha tomado este issue todavía.

Evaluación

Dificultad
4/5
Tiempo estimado
3-5 días
Aptitud para principiantes
35/100
Tipo de issue
Error
Claridad
Bastante claro
Estado de actividad
Estancado
Stack tecnológico
julia
Área
devtools

Línea de trabajo

Comienza con el ejemplo mínimo de Julia del issue y ejecuta la llamada a ReverseDiff.gradient mientras compruebas si se alcanza la ChainRulesCore.rrule personalizada o @scalar_rule. Rastrea la interacción entre ReverseDiff y ChainRules, y considera resuelto el issue cuando se hayan establecido la causa de que se ignoren las reglas y el comportamiento compatible.

Escrito por el modelo de indexación a partir del texto del issue.

Descripción

Hello everyone,

First off, thank you all for your work on ReverseDiff.
I'm not sure whether this issue belongs here on in the ChainRules repo, so forgive me if it's not relevant here.

I have tried defining my own pullback using ChainRules, which claims to be compatible with ReverseDiff.
However, my attempts so far have been unsuccesful.

As a minimal example, take

using ChainRules, ChainRulesCore, ReverseDiff, Plots

function new_gradient(x)
    return conj((abs(x / 2) + 1)^(-2))
end

function H(a)
    return (sign(a) + 1) / 2
end

function ϕ(x)
    return H(x[1])
end

@scalar_rule ϕ(x) new_gradient(x)

function ChainRulesCore.rrule(::typeof(ϕ), x::AbstractArray)
    y = ϕ.(x)

    function array_ϕ_pullback(ȳ)
        println("Test.")
        ϕ̄ = NoTangent()
        x̄ = @thunk new_gradient.(x) * ȳ
        return ϕ̄, x̄
    end

    return y, array_ϕ_pullback
end
gradplot = let
    xs = -5:0.01:5
    vals = [ϕ(x) for x in xs]
    grads = [ReverseDiff.gradient(ϕ, [x]) for x in xs]

    p = plot(xs, vals, label = "ϕ")
    plot!(p, xs, getindex.(grads, 1), label = "∇ϕ")
    p
end

My own rrules are never called.
Have I made an elementary mistake here, or is this actually a problem with ChainRules?

Lenguaje dominante
Julia
Estrellas
397
Forks
62
Merge medio
6 h 46 min
PR fusionados (30 d)
5

Preparar el entorno

Este proyecto no incluye contenedor de desarrollo, Dockerfile ni guía de contribución, así que la configuración corre por tu cuenta: empieza por su README y consulta nuestra guía para la primera contribución para los pasos generales.

Primeros pasos

  1. Lee el issue completo y luego la guía de contribución del proyecto.
  2. Comenta en el issue que vas a ocuparte — evita que dos personas hagan lo mismo.
  3. Haz un fork del repositorio y trabaja en una rama.
  4. Abre un pull request que haga referencia al número del issue.

Más de JuliaDiff/ReverseDiff.jl

Todos los issues de JuliaDiff/ReverseDiff.jl

Issues similares

Más issues de Julia

Recibe los nuevos issues en tu correo

Un resumen breve de issues de GitHub para principiantes.