Fix None graphbolt mask will always be set 0 automatically.

dmlc · Dec 12, 2024 · ecedeae · ecedeae
1 parent 74b2fb5
commit ecedeae
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/python/dgl/distributed/dist_graph.py b/python/dgl/distributed/dist_graph.py
@@ -170,10 +170,7 @@ def process_request(self, server_state):
             gpb = server_state.partition_book
             # Initialize the edge attribute.
             num_edges = g.total_num_edges
-            if self._padding == 0:
-                attr_data = torch.zeros(num_edges, dtype=data_type)
-            else:
-                attr_data = torch.full((num_edges,), self._padding, dtype=data_type)
+            attr_data = torch.full((num_edges,), self._padding, dtype=data_type)
             # Map data from kvstore to the local partition for inner edges only.
             num_inner_edges = gpb.metadata()[gpb.partid]["num_edges"]
             homo_eids = g.edge_attributes[EID][:num_inner_edges]