48c21988cdcde9721cb44804b775d8f0edfcb6cc,tests/nightly/test_np_large_array.py,,test_ctc_loss,#,909

Before Change

    assert type(B).__name__ == "ndarray"
    B.backward()
    assert A.grad.shape == (2, INT_OVERFLOW, 4)
    assert A.grad[0][0][0] == 0

@use_np
def test_erf():
    A = np.ones((2, INT_OVERFLOW))

After Change

    
    L_SEQ, L_ALP, L_LAB, BAT = 2**10, 2**20, 2**6, 2
    A = np.zeros((L_SEQ, BAT, L_ALP))
    label = np.random.randint(0, L_ALP, (BAT, L_LAB))
    // test for expected exception
    test_ctc_loss_size_check(A, label)
    // now we shrink the size a little bit and test for an allowed case
    L_ALP = 2**20 - 1

Instances