dmlc / dmlc/dgl

Dictionary-valued input types for dgl.distributed.dist_graph.node_split

Open
#4,449 5 comments 0 reactions 1 assignee Assigned to @Rhett-Ying View on GitHub
feature request topic: Distributed DGL
Dominant language
Python
Stars
14.3k
Forks
3.1k
PR merge metrics
No merged PRs in 30d

Description

## 🚀 Feature
Dictionary-valued input types for dgl.distributed.dist_graph.node_split

## Motivation

Currently, node split only accepts a 1D tensor for nodes argument. In cases when you're working with a heterogenous graph, the split (if stored on graph) likely comes in a dictionary. Adding dictionary-typed inputs removes user friction.

## Pitch

Add a logic path accepting dicts as input for nodes, performing the current set of functionality extrapolated to a heterogenous dictionary.

Contributor guide

No contributing guide indexed for this repository

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.