-
Notifications
You must be signed in to change notification settings - Fork 8
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
conv
gradient is not implemented in EnzymeJAX
#214
Comments
the feature_group_count is probably missing on EnzymeJAX end? cc @Pangoraw |
The reverse is not implemented for convolution. It should be fixed in Enzyme-JAX |
Oh ok I will wait for doing my MISNT benchmark then, have a nice day |
@Pangoraw do you want to bump the commits of EnzymeJaX and reactant and bump the jll Setup needs to bump: Reactant.jl/deps/ReactantExtra/WORKSPACE Line 12 in a17315c
Then https://github.com/JuliaPackaging/Yggdrasil/blob/ff9587a0df76e20171efd10076a59395c4fad5dd/R/Reactant/build_tarballs.jl#L12 |
conv
gradient is not implemented in EnzymeJAX
@Pangoraw @mofeing I don't think this is implemented in EnzymeJaX atm: https://github.com/EnzymeAD/Enzyme-JAX/blob/main/test/lit_tests/diffrules/stablehlo/convolution.mlir , no? |
I meant that there is a rule in https://github.com/EnzymeAD/Enzyme-JAX/blob/724be0666e0f9fbba66f6b2fddbdb222b0db5fb6/src/enzyme_ad/jax/Implementations/HLODerivatives.td#L588 but it generates invalid code instead of returning an unimplemented error. |
Ah I read your comment of “it should be fixed In the other repo” as it is fixed but needs a jll bump rather than your intended, we should fix it but the bug is in another repo |
This is fixed in the latest release |
Hello, I wonder if Reactant works with Conv layers, it seems it works in forward but not in the gradient pass, neither on cpu of gpu
version :
code :
error :
The text was updated successfully, but these errors were encountered: