[ci skip]

setup.py

@@ -31,6 +31,7 @@ def get_extensions():
     extensions_dir = osp.join('csrc')
     main_files = glob.glob(osp.join(extensions_dir, '*.cpp'))
     main_files = ['csrc/version.cpp', 'csrc/metis.cpp']
+    main_files = ['csrc/version.cpp']
 
     for main, suffix in product(main_files, suffices):
         define_macros = []

torch_sparse/__init__.py

@@ -8,17 +8,8 @@ __version__ = '0.6.10'
 suffix = 'cuda' if torch.cuda.is_available() else 'cpu'
 
 for library in [
-        '_version',
-        # '_convert',
-        # '_diag',
-        # '_spmm',
-        # '_spspmm',
-        '_metis',
-        # '_rw',
-        # '_saint',
-        # '_sample',
-        # '_ego_sample',
-        # '_relabel',
+        '_version', '_convert', '_diag', '_spmm', '_spspmm', '_metis', '_rw',
+        '_saint', '_sample', '_ego_sample', '_relabel'
 ]:
     torch.ops.load_library(importlib.machinery.PathFinder().find_spec(
         f'{library}_{suffix}', [osp.dirname(__file__)]).origin)