Skip to content

check compatibility with Mooncake on CPU #640

Description

@CarloLucibello

Check that all our layers and operators are compatible with Mooncake.jl AD

Activity

  1. Parvm1102 commented on Mar 2, 2026

    @Parvm1102
    Contributor

    Hi @CarloLucibello ,

    I have done some progress on this: I've set up the test suite for the spatial convolutional layers comparing Mooncake and Zygote.

    Results: 16 / 20 conv layers are compatible (935 passed tests).
    However, there are errors with these two layers:

    • DConv (3 errors): Fails during reverse-mode differentiation (might be related to Wrong implementation DConv #592).
    • ChebConv (1 error): Crashes during gradient cache preparation before the test executes.

    2 layers were excluded: CGConv and GMMConv, same as existing tests.

    How would you like me to proceed? Should I dig into finding workarounds for DConv and ChebConv, or should I move on to testing the other layer categories (pool.jl, heteroconv.jl, etc.)? I can also open a PR if you want to see the code.

  2. CarloLucibello commented on Mar 3, 2026

    @CarloLucibello
    MemberAuthor

    Hi @Parvm1102, thanks.
    I suggest you implement something similar to the test infrastructure of Flux (see the test_gradients function)
    https://github.com/FluxML/Flux.jl/blob/master/test/test_utils.jl#L82

    Then open a PR with just the conv layers tested and we will discuss there.

  3. Parvm1102 commented on Mar 5, 2026

    @Parvm1102
    Contributor

    Currently incompatible layers (found in #641 and #642 ):

    • ChebConv
    • DConv
    • GNNChain
    • GConvLSTMCell / GConvLSTM
    • GConvGRUCell / GConvGRU
    • DCGRUCell/ DCGRU
    • EvolveGCNOCell
  4. muskan-101 commented on Mar 18, 2026

    @muskan-101

    @CarloLucibello Hey, Can I work on this issue I have theoretical knowledge of Neural Networks and a little hands on experience as well, we were taught Neural Networks and fuzzy logic as a separate course in our college. So should I work on this or is there any issue that will be more suitable for me. I would really appreciate your suggestions.

  5. jeanphilippeadielou commented on Mar 27, 2026

    @jeanphilippeadielou

    Hi, I'm willing to have a look. I got a few basis in Julia and got a good understanding of Neural Networks, as well as Graphs/Networks more broadly on the theoretical side.

  6. CarloLucibello commented on Jul 31, 2026

    @CarloLucibello
    MemberAuthor

    @Parvm1102 this issue should be updated with the current status

  7. changed the title [-]check compatibility with Mooncake[/-] [+]check compatibility with Mooncake on CPU[/+] on Jul 31, 2026
  8. CarloLucibello commented on Jul 31, 2026

    @CarloLucibello
    MemberAuthor

    let's keep track of CUDA compatibility in #702 and leave this issue for CPU

  9. jeanphilippeadielou commented on Aug 13, 2026

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

Metadata

Metadata

Assignees

No one assigned

    Labels

    Type

    No type

    Projects

    No projects

      Milestone

      No milestone

      Relationships

      None yet

      Development

      No branches or pull requests

      Issue actions