test_addone.py 328 Bytes
Newer Older
nhynes committed
1 2 3 4 5 6 7 8 9 10 11 12 13
import tvm
import numpy as np

ctx = tvm.context('cpu', 0)
fadd1 = tvm.module.load('lib/test_addone.signed.so')

n = 10
x = tvm.nd.array(np.random.uniform(size=n).astype('float32'), ctx)
y = tvm.nd.array(np.zeros(n, dtype='float32'), ctx)
fadd1(x, y)

np.testing.assert_allclose(y.asnumpy(), x.asnumpy() + 1)
print("It works!")