()
| 148 | |
| 149 | @pytest.mark.skipif(not env.has_rpc(), reason="need rpc") |
| 150 | def test_rpc_large_array(): |
| 151 | # testcase of large array creation |
| 152 | server = rpc.Server() |
| 153 | remote = rpc.connect("127.0.0.1", server.port) |
| 154 | |
| 155 | def check_remote(): |
| 156 | dev = remote.cpu(0) |
| 157 | a_np = np.ones((5041, 720)).astype("float32") |
| 158 | b_np = np.ones((720, 192)).astype("float32") |
| 159 | a = tvm.runtime.tensor(a_np, dev) |
| 160 | b = tvm.runtime.tensor(b_np, dev) |
| 161 | np.testing.assert_equal(a.numpy(), a_np) |
| 162 | np.testing.assert_equal(b.numpy(), b_np) |
| 163 | |
| 164 | check_remote() |
| 165 | |
| 166 | |
| 167 | @tvm.testing.skip_if_32bit(reason="skipping test for i386.") |
nothing calls this directly
no test coverage detected
searching dependent graphs…