Signed-off-by: rshaw@neuralmagic.com <robertgshaw2@gmail.com>
This commit is contained in:
rshaw@neuralmagic.com 2025-06-27 15:12:03 +00:00
parent fea0731cf4
commit 5d2eac70e7

View File

@ -921,15 +921,6 @@ class NixlConnectorWorker:
# corresponding rank. With heterogeneous TP, fixing D>P, the D tp
# workers will issue xfers to parts of the P worker remote kv caches.
# # Sort block ids to ensure nixl can merge contiguous blocks.
# start = time.perf_counter()
# sorted_idx = sorted(range(len(local_block_ids)),
# key=local_block_ids.__getitem__)
# local_block_ids = [local_block_ids[i] for i in sorted_idx]
# remote_block_ids = [remote_block_ids[i] for i in sorted_idx]
# end = time.perf_counter()
# print(f"REORDER took: {end - start}")
# Get descs ids.
local_block_descs_ids: list[int] = []
remote_block_descs_ids: list[int] = []