Function verify

tests/python/contrib/test_hipblas.py:36–48

(target="rocm")

    C = hipblas.matmul(A, B, dtype=out_dtype)

    def verify(target="rocm"):
        if not tvm.get_global_func("tvm.contrib.hipblas.matmul", True):
            print("skip because extern function is not available")
            return
        dev = tvm.rocm(0)
        f = tvm.compile(te.create_prim_func([A, B, C]), target=target)
        a = tvm.runtime.tensor(np.random.uniform(0, 128, size=(n, l)).astype(A.dtype), dev)
        b = tvm.runtime.tensor(np.random.uniform(0, 128, size=(l, m)).astype(B.dtype), dev)
        c = tvm.runtime.tensor(np.zeros((n, m), dtype=C.dtype), dev)
        f(a, b, c)
        tvm.testing.assert_allclose(
            c.numpy(), np.dot(a.numpy().astype(C.dtype), b.numpy().astype(C.dtype)), rtol=rtol
        )

    verify()