Method t

python/tests/test_array.py:2229–2233

()

        key = mx.array([1, 2])

        def t():
            a, b = mx.split(key, 2)
            a = mx.reshape(a, [])
            b = mx.reshape(b, [])
            return b

        mx.synchronize()
        t()