.DO NOT MERGE: Find issue with CAReduce

Author: ricardoV94

tests/link/mlx/test_elemwise.py

@@ -42,11 +42,14 @@
 def test_input(op) -> None:
     x = vector("x")
     out = op(x > 0)
+    out.dprint(print_type=True)
+    assert 0
     x_test = mx.array([1.0, 2.0, 3.0])
 
     compare_mlx_and_py([x], out, [x_test])
 
 
+@pytest.mark.skip(reason="It's crashing the kernel")
 def test_mlx_CAReduce():
     a_pt = vector("a")
     a_pt.tag.test_value = np.r_[1, 2, 3].astype(config.floatX)
@@ -78,6 +81,7 @@ def test_mlx_CAReduce():
     compare_mlx_and_py([a_pt], [x], [np.c_[[1, 2, 3], [1, 2, 3]].astype(config.floatX)])
 
 
+@pytest.mark.skip(reason="It's crashing the kernel")
 @pytest.mark.parametrize("axis", [None, 0, 1])
 def test_softmax(axis):
     x = matrix("x")
@@ -86,6 +90,7 @@ def test_softmax(axis):
     compare_mlx_and_py([x], [out], [x_test_value])
 
 
+@pytest.mark.skip(reason="It's crashing the kernel")
 @pytest.mark.parametrize("axis", [None, 0, 1])
 def test_softmax_grad(axis):
     dy = matrix("dy")
@@ -97,6 +102,7 @@ def test_softmax_grad(axis):
     compare_mlx_and_py([dy, sm], [out], [dy_test_value, sm_test_value])
 
 
+@pytest.mark.skip(reason="It's crashing the kernel")
 @pytest.mark.parametrize("axis", [None, 0, 1])
 def test_logsoftmax(axis):
     x = matrix("x")