Function verify

tests/python/contrib/test_random.py:38–52

(target="llvm")

    A = random.randint(-127, 128, size=(m, n), dtype="int32")

    def verify(target="llvm"):
        if not tvm.testing.device_enabled(target):
            print(f"skip because {target} is not enabled...")
            return
        if not tvm.get_global_func("tvm.contrib.random.randint", True):
            print("skip because extern function is not available")
            return
        dev = tvm.cpu(0)
        f = tvm.compile(te.create_prim_func([A]), target=target)
        a = tvm.runtime.tensor(np.zeros((m, n), dtype=A.dtype), dev)
        f(a)
        na = a.numpy()
        assert abs(np.mean(na)) < 0.3
        assert np.min(na) == -127
        assert np.max(na) == 127

    verify()