fix: conv_general differences between gpu, cpu (#2070)

* fix general_conv padding * fix bugs * add test --------- Co-authored-by: Awni Hannun <awni@apple.com>
2025-12-16 01:49:05 +08:00 · 2025-05-09 20:26:52 +03:00
parent 0cae0bdac8
commit a7fae8a176
6 changed files with 413 additions and 270 deletions
--- a/mlx/ops.cpp
+++ b/mlx/ops.cpp
@@ -3974,6 +3974,7 @@ array conv_general(
          to_stream(s),
          stride,
          padding_lo,
+          padding_hi,
          kernel_dilation,
          input_dilation,
          groups,