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

Function remote

tests/python/nightly/test_nnapi/conftest.py:25–35  ·  view source on GitHub ↗
()

Source from the content-addressed store, hash-verified

23
24
25def remote():
26 required_env = ("TVM_TRACKER_HOST", "TVM_TRACKER_PORT", "RPC_DEVICE_KEY")
27 missing = [name for name in required_env if name not in os.environ]
28 if missing:
29 pytest.skip(f"NNAPI remote environment unavailable: {', '.join(missing)} not set")
30 rpc_tracker_host = os.environ["TVM_TRACKER_HOST"]
31 rpc_tracker_port = int(os.environ["TVM_TRACKER_PORT"])
32 rpc_device_key = os.environ["RPC_DEVICE_KEY"]
33 tracker = rpc.connect_tracker(rpc_tracker_host, rpc_tracker_port)
34 remote = tracker.request(rpc_device_key, priority=0, session_timeout=600)
35 return remote, tracker

Callers 10

test_networkFunction · 0.90
test_unaryFunction · 0.90
test_elementwise_binaryFunction · 0.90
test_divideFunction · 0.90
test_matmulFunction · 0.90
test_permute_dimsFunction · 0.90
test_astypeFunction · 0.90
test_meanFunction · 0.90
test_conv2dFunction · 0.90
test_max_pool2dFunction · 0.90

Calls 3

connect_trackerMethod · 0.80
joinMethod · 0.45
requestMethod · 0.45

Tested by

no test coverage detected

Used in the wild real call sites across dependent graphs

searching dependent graphs…