MCPcopy Create free account
hub / github.com/apache/tvm / test_rpc_tracker_via_proxy

Function test_rpc_tracker_via_proxy

tests/python/runtime/test_runtime_rpc.py:592–629  ·  view source on GitHub ↗

tracker / \ Host -- Proxy -- RPC server

(device_key)

Source from the content-addressed store, hash-verified

590@pytest.mark.skipif(not env.has_rpc(), reason="need rpc")
591@pytest.mark.parametrize("device_key", ["test_device", "127.0.0.1:5555"])
592def test_rpc_tracker_via_proxy(device_key):
593 """
594 tracker
595 / \
596 Host -- Proxy -- RPC server
597 """
598
599 tracker_server = Tracker(port=9000, port_end=9100)
600 proxy_server = Proxy(
601 host=tracker_server.host,
602 port=8888,
603 port_end=8988,
604 tracker_addr=(tracker_server.host, tracker_server.port),
605 )
606
607 server1 = rpc.Server(
608 host=proxy_server.host,
609 port=proxy_server.port,
610 key=device_key,
611 tracker_addr=(tracker_server.host, tracker_server.port),
612 is_proxy=True,
613 )
614 server2 = rpc.Server(
615 host=proxy_server.host,
616 port=proxy_server.port,
617 key=device_key,
618 tracker_addr=(tracker_server.host, tracker_server.port),
619 is_proxy=True,
620 )
621
622 client = rpc.connect_tracker(tracker_server.host, tracker_server.port)
623 remote1 = client.request(device_key, session_timeout=30) # pylint: disable=unused-variable
624 remote2 = client.request(device_key, session_timeout=30) # pylint: disable=unused-variable
625
626 server2.terminate()
627 server1.terminate()
628 proxy_server.terminate()
629 tracker_server.terminate()
630
631
632@pytest.mark.skipif(not env.has_rpc(), reason="need rpc")

Callers

nothing calls this directly

Calls 8

terminateMethod · 0.95
terminateMethod · 0.95
terminateMethod · 0.95
TrackerClass · 0.90
ProxyClass · 0.90
ServerMethod · 0.80
connect_trackerMethod · 0.80
requestMethod · 0.45

Tested by

no test coverage detected

Used in the wild real call sites across dependent graphs

searching dependent graphs…