MCPcopy Index your code
hub / github.com/apache/tvm / test_int

Function test_int

tests/python/disco/test_session.py:156–162  ·  view source on GitHub ↗
(session_kind)

Source from the content-addressed store, hash-verified

154
155@pytest.mark.parametrize("session_kind", _all_session_kinds)
156def test_int(session_kind): # pylint: disable=invalid-name
157 num_workers = 4
158 sess = session_kind(num_workers=num_workers)
159 func: di.DPackedFunc = sess.get_global_func("tests.disco.add_one")
160 result: di.DRef = func(1)
161 for i in range(num_workers):
162 assert result.debug_get_from_remote(i) == 2
163
164
165@pytest.mark.parametrize("session_kind", _all_session_kinds)

Callers

nothing calls this directly

Calls 3

get_global_funcMethod · 0.80
debug_get_from_remoteMethod · 0.80
funcFunction · 0.50

Tested by

no test coverage detected

Used in the wild real call sites across dependent graphs

searching dependent graphs…