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

Function result_func

python/tvm/runtime/disco/process_pool.py:185–191  ·  view source on GitHub ↗
(worker_id: int)

Source from the content-addressed store, hash-verified

183 pool = [DiscoPopenWorker(i, num_workers, num_groups, entrypoint) for i in range(1, num_workers)]
184
185 def result_func(worker_id: int):
186 nonlocal pool
187 if worker_id != 0:
188 read_fd, write_fd = pool[worker_id - 1].start()
189 return Shape([read_fd, write_fd])
190 del pool
191 return None
192
193 return result_func

Callers

nothing calls this directly

Calls 2

ShapeFunction · 0.50
startMethod · 0.45

Tested by

no test coverage detected

Used in the wild real call sites across dependent graphs

searching dependent graphs…