Unverified Commit 799245a7 authored by Serge Panev's avatar Serge Panev Committed by GitHub
Browse files

[Dist] Fix bug in Dist partitioning (#4910)


Signed-off-by: default avatarSerge Panev <spanev@nvidia.com>
Signed-off-by: default avatarSerge Panev <spanev@nvidia.com>
parent 53895b95
...@@ -181,7 +181,7 @@ class DistLookupService: ...@@ -181,7 +181,7 @@ class DistLookupService:
continue continue
# Get the node_type_ids and per_type_nids for the incoming global_nids. # Get the node_type_ids and per_type_nids for the incoming global_nids.
ntype_ids, type_nids = self.id_map(owner_req_list[idx].numpy()) ntype_ids, type_nids = self.id_map(owner_req_list[idx].numpy())
nypte_ids, type_nids = ntype_ids.numpy(), type_nids.numpy() ntype_ids, type_nids = ntype_ids.numpy(), type_nids.numpy()
# Lists to store partition-ids for the incoming global-nids. # Lists to store partition-ids for the incoming global-nids.
type_id_lookups = [] type_id_lookups = []
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment