[Bugfix] fix a potential context mismatch (#154)

ed1854e8 · Gan Quan · Minjie Wang · 53b9a4bd · ed1854e8
Commit ed1854e8 authored Nov 15, 2018 by Gan Quan Committed by Minjie Wang Nov 15, 2018
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 2 deletions

python/dgl/scheduler.py python/dgl/scheduler.py +3 -2

No files found.
--- a/python/dgl/scheduler.py
+++ b/python/dgl/scheduler.py
@@ -348,8 +348,9 @@ class SendRecvExecutor(BasicExecutor):
                dat = F.squeeze(dat, dim=1)
        else:
            dat = F.ones((len(self.u), ), dtype=F.float32, ctx=ctx)
-        adjmat = F.sparse_matrix(dat, ('coo', self.graph_idx), self.graph_shape)
-        return F.copy_to(adjmat, ctx)
+        adjmat = F.sparse_matrix(
+                dat, ('coo', F.copy_to(self.graph_idx, ctx)), self.graph_shape)
+        return adjmat


 class BundledExecutor(BasicExecutor):