95f5cc60904a2d88d4861fff0f6dbad15f8cdbe3,tests/nightly/test_np_large_array.py,,test_amax,#,187

Before Change

    A[100][1] = 1
    A.attach_grad()
    with mx.autograd.record():
        B = np.amax(A)
    print(B)
    assert B == 1.0
    B.backward()
    assert A.grad.shape == (INT_OVERFLOW, 2)
    assert A.grad[0][0] == 0

After Change

        out.backward()
    assert out == 1.0
    assert inp.grad.shape == (INT_OVERFLOW, 2)
    assert inp.grad[0, 0] == 0 and inp.grad[-1, -1] == 1


@use_np
def test_argmin():

Instances