Skip to content

T
tic
Commit 2e07447e by Tianqi Chen Committed by Yizhi Liu
Options

[TOPI] Fix flaky testcase for check round (#4211)

parent 780a9945
Showing with 2 additions and 2 deletions
topi/tests/python/test_topi_math.py
......@@ -52,7 +52,7 @@ def test_ewise():
a_np = np.random.uniform(low=low, high=high, size=shape).astype(A.dtype) * 10
# avoid round check too close to boundary
if check_round:
a_np += ((np.fmod(a_np, 1) - 0.5) < 1e-6) * 1e-5
a_np += ((np.abs(np.fmod(a_np, 1)) - 0.5) < 1e-6) * 1e-5
b_np = f_numpy(a_np)
def check_device(device):
......@@ -100,7 +100,7 @@ def test_ewise():
a_np.ravel()[np.random.choice(a_np.size, int(a_np.size * 0.5), replace=False)] = np.nan
# avoid round check too close to boundary
if check_round:
a_np += ((np.fmod(a_np, 1) - 0.5) < 1e-6) * 1e-5
a_np += ((np.abs(np.fmod(a_np, 1)) - 0.5) < 1e-6) * 1e-5
b_np = np.isnan(a_np)
def check_device(device):
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment