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

Function test_init

tests/python/disco/test_ccl.py:52–55  ·  view source on GitHub ↗
(session_kind, ccl)

Source from the content-addressed store, hash-verified

50@pytest.mark.parametrize("session_kind", _all_session_kinds)
51@pytest.mark.parametrize("ccl", _ccl)
52def test_init(session_kind, ccl):
53 devices = [0, 1]
54 sess = session_kind(num_workers=len(devices))
55 sess.init_ccl(ccl, *devices)
56
57
58@pytest.mark.parametrize("session_kind", _all_session_kinds)

Callers

nothing calls this directly

Calls 1

init_cclMethod · 0.80

Tested by

no test coverage detected

Used in the wild real call sites across dependent graphs

searching dependent graphs…