def test_gufunc(self):
gufunc = GUVectorize(matmulcore, '(m,n),(n,p)->(m,p)',
target=self.target)
gufunc.add((float32[:, :], float32[:, :], float32[:, :]))
gufunc = gufunc.build_ufunc()
self.check_matmul_gufunc(gufunc)
def test_ufunc_like(self):
# Test problem that the stride of "scalar" gufunc argument not properly
# handled when the actual argument is an array,
# causing the same value (first value) being repeated.
gufunc = GUVectorize(axpy, '(), (), () -> ()', target=self.target)
gufunc.add('(intp, intp, intp, intp[:])')
gufunc = gufunc.build_ufunc()
x = np.arange(10, dtype=np.intp)
out = gufunc(x, x, x)
np.testing.assert_equal(out, x * x + x)
def test_dynamic_matmul(self):
def check_matmul_gufunc(gufunc, A, B, C):
Gold = ut.matrix_multiply(A, B)
gufunc(A, B, C)
np.testing.assert_allclose(C, Gold, rtol=1e-5, atol=1e-8)
gufunc = GUVectorize(matmulcore,
'(m,n),(n,p)->(m,p)',
target=self.target,
is_dynamic=True)
matrix_ct = 10
Ai64 = np.arange(matrix_ct * 2 * 4,
dtype=np.int64).reshape(matrix_ct, 2, 4)
Bi64 = np.arange(matrix_ct * 4 * 5,
dtype=np.int64).reshape(matrix_ct, 4, 5)
Ci64 = np.arange(matrix_ct * 2 * 5,
dtype=np.int64).reshape(matrix_ct, 2, 5)
check_matmul_gufunc(gufunc, Ai64, Bi64, Ci64)
A = np.arange(matrix_ct * 2 * 4,
dtype=np.float32).reshape(matrix_ct, 2, 4)
B = np.arange(matrix_ct * 4 * 5,
dtype=np.float32).reshape(matrix_ct, 4, 5)
C = np.arange(matrix_ct * 2 * 5,
dtype=np.float32).reshape(matrix_ct, 2, 5)
check_matmul_gufunc(gufunc, A, B, C) # trigger compilation
self.assertEqual(len(gufunc.types), 2) # ensure two versions of gufunc
def test_dynamic_ufunc_like(self):
def check_ufunc_output(gufunc, x):
out = np.zeros(10, dtype=x.dtype)
gufunc(x, x, x, out)
np.testing.assert_equal(out, x * x + x)
# Test problem that the stride of "scalar" gufunc argument not properly
# handled when the actual argument is an array,
# causing the same value (first value) being repeated.
gufunc = GUVectorize(axpy,
'(), (), () -> ()',
target=self.target,
is_dynamic=True)
x = np.arange(10, dtype=np.intp)
check_ufunc_output(gufunc, x)