Function assertTensorEqual

test/python/test_opt.py:32–44

(x,y,decimal=6)

from cuda_helper import gpu_dev, cpu_dev

def assertTensorEqual(x,y,decimal=6):
    assert x.shape == y.shape
    assert x.dtype == y.dtype
    assert x.device.id() == y.device.id()
    d = x.device
    x.to_host()
    y.to_host()
    np.testing.assert_array_almost_equal(
        x.data.GetFloatValue(int(x.size())),
        y.data.GetFloatValue(int(y.size())),
                                    decimal)
    x.to_device(d)
    y.to_device(d)

def on_cpu_gpu(func):
    @functools.wraps(func)