zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-12-16 01:49:05 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

[CUDA] Fix conv grads with groups

Browse Source
This commit is contained in:
Cheng
2025-08-14 03:21:00 -07:00
parent 8bf8034ffd
commit 5b05aaad95
3 changed files with 44 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
python/tests/cuda_skip.py
View File

@@ -17,7 +17,6 @@ cuda_skip = {
"TestConv.test_1d_conv_with_2d",
"TestConv.test_conv_1d_groups_flipped",
"TestConv.test_conv_general_flip_grad",
"TestConv.test_conv_groups_grad",
"TestConv.test_torch_conv_2D",
"TestConv.test_torch_conv_depthwise",
"TestConv.test_torch_conv_general",