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
...@@ -274,11 +274,8 @@ def message_passing(g, mfunc, rfunc, afunc): ...@@ -274,11 +274,8 @@ 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. getattr(ops, '{}_{}'.format(mfunc.name, rfunc.name), None) is not None):
ndata = {}
elif (is_builtin(mfunc) and is_builtin(rfunc) and
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)
else: else:
......
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