Skip to content

Commit 0d40493

Browse files
committed
Remove use of Numpy functions from test_tensor_accumulation and increase size of test_logcumsumexp_basic
1 parent 1b3ade0 commit 0d40493

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

dpctl/tests/test_tensor_accumulation.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,6 @@
1616

1717
from random import randrange
1818

19-
import numpy as np
2019
import pytest
2120
from helper import get_queue_or_skip, skip_if_dtype_not_supported
2221

@@ -376,13 +375,13 @@ def test_logcumsumexp_basic():
376375
get_queue_or_skip()
377376

378377
dt = dpt.float32
379-
x = dpt.ones(10, dtype=dt)
378+
x = dpt.ones(1000, dtype=dt)
380379
r = dpt.cumulative_logsumexp(x)
381380

382-
expected = 1 + np.log(np.arange(1, 11, dtype=dt))
381+
expected = 1 + dpt.log(dpt.arange(1, 1001, dtype=dt))
383382

384383
tol = 4 * dpt.finfo(dt).resolution
385-
assert np.allclose(dpt.asnumpy(r), expected, atol=tol, rtol=tol)
384+
assert dpt.allclose(r, expected, atol=tol, rtol=tol)
386385

387386

388387
def geometric_series_closed_form(n, dtype=None, device=None):

0 commit comments

Comments
 (0)