Skip to content

GitLab

"git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "8b1e3ec93e8e4d27d56b707aec794abc44334e34"
Unverified Commit 98e11533 authored by Zihao Ye's avatar Zihao Ye Committed by GitHub
Browse files

[hotfix] Produce empty tensor in data frame for emtpy graphs. (#2387) 

* upd

* lint

* lint1
parent 1a932926
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 5 deletions
python/dgl/core.py
View file @ 98e11533
...@@ -274,10 +274,7 @@ def message_passing(g, mfunc, rfunc, afunc): ...@@ -274,10 +274,7 @@ def message_passing(g, mfunc, rfunc, afunc):
dict[str, Tensor] dict[str, Tensor]
Results from the message passing computation. Results from the message passing computation.
""" """
if g.number_of_edges() == 0: if (is_builtin(mfunc) and is_builtin(rfunc) and
# No message passing is triggered.
ndata = {}
elif (is_builtin(mfunc) and is_builtin(rfunc) and
getattr(ops, '{}_{}'.format(mfunc.name, rfunc.name), None) is not None): getattr(ops, '{}_{}'.format(mfunc.name, rfunc.name), None) is not None):
# invoke fused message passing # invoke fused message passing
ndata = invoke_gspmm(g, mfunc, rfunc) ndata = invoke_gspmm(g, mfunc, rfunc)
......
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