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

Function test_float

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

Source from the content-addressed store, hash-verified

164
165@pytest.mark.parametrize("session_kind", _all_session_kinds)
166def test_float(session_kind):
167 num_workers = 4
168 sess = session_kind(num_workers=num_workers)
169 func: di.DPackedFunc = sess.get_global_func("tests.disco.add_one_float")
170 result: di.DRef = func(1.5)
171
172 for i in range(num_workers):
173 assert result.debug_get_from_remote(i) == 2.0
174
175
176@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…