[Metal] Add implicit matmul pathway for mx.conv3d (#3147)
Co-authored-by: Gleb Sterkin <[email protected]> Co-authored-by: Angelos Katharopoulos <[email protected]>
This commit is contained in:
co-authored by
Gleb Sterkin
Angelos Katharopoulos
parent
d4c81062ad
commit
1d8d693d08
@@ -550,7 +550,6 @@ class TestConv(mlx_tests.MLXTestCase):
|
||||
(1, 1, 6),
|
||||
(4, 16, 32),
|
||||
):
|
||||
continue
|
||||
for idim, kdim, stride, padding in (
|
||||
((1, 1, 1), (1, 1, 1), (1, 1, 1), (0, 0, 0)),
|
||||
((3, 3, 3), (3, 1, 1), (1, 1, 1), (0, 0, 0)),
|
||||
|
||||
Reference in New Issue
Block a user