in python/featgraph/module/sddmm.py [0:0]
def edge_mapping(self):
assert self._target != 'cuda', "no graph partitioning on cuda, no edge_mapping"
assert self._num_row_partitions > 1 or self._num_col_partitions > 1, \
"no graph partitioning, no edge_mapping"
return self._edge_mapping