[Testing] Utility method to run TVM on remote device

[junrushao]

This PR introduces tvm.testing.rpc_run, a utility method that allows a runtime.Module to run on a remote device via TVM RPC.

Example:

import numpy as np
import tvm
from tvm.script import tir as T
from tvm.testing import rpc_run

@T.prim_func
def cuda_kernel(
    A: T.Buffer((128,), "float32"),
    B: T.Buffer((128,), "float32"),
):
    for bx in T.thread_binding(4, thread="blockIdx.x"):
        for tx in T.thread_binding(32, thread="threadIdx.x"):
            x = bx * 32 + tx
            B[x] = A[x] + 1.0

def main():
    np_a = np.random.randn(128).astype("float32")
    np_b = np_a + 1.0

    rt_mod = tvm.build(cuda_kernel, target="nvidia/geforce-rtx-3090-ti")
    tvm_a, tvm_b = rpc_run(
        rt_mod,
        "cuda",
        [np_a, np_b],
    )

    assert np.allclose(tvm_b, np_b)

Result:

Execution time summary:
 mean (ms)   median (ms)    max (ms)     min (ms)     std (ms)
  0.0023       0.0023       0.0023       0.0023       0.0000

[tvm-bot]

Thanks for contributing to TVM! Please refer to the contributing guidelines https://tvm.apache.org/docs/contribute/ for useful information and tips. Please request code reviews from Reviewers by @-ing them in a comment.

• No users to tag found in teams: testing _{See #10317 for details}

_{Generated by tvm-bot}

[MasterJH5574]

Thanks!