def METHOD_NAME():
    def matmul(x, w):
        return torch.matmul(x, w)
    x = torch.randn(15, 20)
    w = torch.randn(20, 30)
    example_inputs = (x, w)
    rt_mod = optimize_torch(matmul, example_inputs)
    torch_answer = torch.matmul(x, w).numpy()
    tvm_answer = rt_mod(x, w).numpy()
    tvm.testing.assert_allclose(torch_answer, tvm_answer, atol=1e-5, rtol=1e-5)