fix conv export (#2265)

2025-09-17 17:28:10 +08:00 · 2025-06-10 09:34:01 -07:00
parent 7c4eb5d03e
commit 62fecf3e13
2 changed files with 35 additions and 1 deletions
--- a/mlx/primitives.h
+++ b/mlx/primitives.h
@@ -719,9 +719,9 @@ class Convolution : public UnaryPrimitive {
  bool is_equivalent(const Primitive& other) const override;
  auto state() const {
    return std::make_tuple(
+        kernel_strides_,
        padding_lo_,
        padding_hi_,
-        kernel_strides_,
        kernel_dilation_,
        input_dilation_,
        groups_,