We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent bb87ec6 commit d0f5005Copy full SHA for d0f5005
torch_scatter/__init__.py
@@ -1,6 +1,8 @@
1
+import os.path as osp
2
+
3
import torch
4
-torch.ops.load_library('torch_scatter/_C.so')
5
+torch.ops.load_library(osp.join(osp.dirname(osp.abspath(__file__)), '_C.so'))
6
7
from .segment_csr import (segment_sum_csr, segment_add_csr, segment_mean_csr,
8
segment_min_csr, segment_max_csr, segment_csr,
0 commit comments