def call_bfs(sID, data, num_verts, num_edges, vertex_partition_offsets, start,
return_distances):
wid = Comms.get_worker_id(sID)
handle = Comms.get_handle(sID)
return mg_bfs.mg_bfs(data[0], num_verts, num_edges,
vertex_partition_offsets, wid, handle, start,
return_distances)
def call_bfs(sID, data, num_verts, num_edges, vertex_partition_offsets,
aggregate_segment_offsets, start, depth_limit, return_distances):
wid = Comms.get_worker_id(sID)
handle = Comms.get_handle(sID)
local_size = len(aggregate_segment_offsets) // Comms.get_n_workers(sID)
segment_offsets = \
aggregate_segment_offsets[local_size * wid: local_size * (wid + 1)]
return mg_bfs.mg_bfs(data[0], num_verts, num_edges,
vertex_partition_offsets, wid, handle,
segment_offsets, start, depth_limit, return_distances)
def call_bfs(sID, data, local_data, start, return_distances):
wid = Comms.get_worker_id(sID)
handle = Comms.get_handle(sID)
return mg_bfs.mg_bfs(data[0], local_data, wid, handle, start,
return_distances)