mlx
mlx copied to clipboard
[BUG] CUDA conv test hangs on B200
python python/tests/test_conv.py TestConv.test_torch_conv_2D
It hangs in one of the grouped conv tests.