Skip to content

GitLab

Unverified Commit be53add4 authored by Zihao Ye's avatar Zihao Ye Committed by GitHub
Browse files

relax (#1802)

parent bc209bff
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
tests/compute/test_sparse.py
View file @ be53add4
...@@ -142,14 +142,14 @@ def test_spmm(g, shp, msg, reducer, index_dtype): ...@@ -142,14 +142,14 @@ def test_spmm(g, shp, msg, reducer, index_dtype):
if reducer in ['min', 'max']: # there might be some numerical errors if reducer in ['min', 'max']: # there might be some numerical errors
rate = F.reduce_sum(F.abs(F.grad(g.srcdata['x']) - grad_u)) /\ rate = F.reduce_sum(F.abs(F.grad(g.srcdata['x']) - grad_u)) /\
F.reduce_sum(F.abs(grad_u)) F.reduce_sum(F.abs(grad_u))
assert F.as_scalar(rate) < 1e-3, rate assert F.as_scalar(rate) < 1e-2, rate
else: else:
assert F.allclose(F.grad(g.srcdata['x']), grad_u) assert F.allclose(F.grad(g.srcdata['x']), grad_u)
if msg != 'copy_lhs': if msg != 'copy_lhs':
if reducer in ['min', 'max']: if reducer in ['min', 'max']:
rate = F.reduce_sum(F.abs(F.grad(g.edata['w']) - grad_e)) /\ rate = F.reduce_sum(F.abs(F.grad(g.edata['w']) - grad_e)) /\
F.reduce_sum(F.abs(grad_e)) F.reduce_sum(F.abs(grad_e))
assert F.as_scalar(rate) < 1e-3, rate assert F.as_scalar(rate) < 1e-2, rate
else: else:
assert F.allclose(F.grad(g.edata['w']), grad_e) assert F.allclose(F.grad(g.edata['w']), grad_e)
print('backward passed') print('backward passed')
......
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