Skip to content

GitLab

Commit 4833b184 authored by rusty1s's avatar rusty1s
Browse files

[ci skip]

parent f342db64
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 11 deletions
setup.py
View file @ 4833b184
...@@ -31,6 +31,7 @@ def get_extensions(): ...@@ -31,6 +31,7 @@ def get_extensions():
extensions_dir = osp.join('csrc') extensions_dir = osp.join('csrc')
main_files = glob.glob(osp.join(extensions_dir, '*.cpp')) main_files = glob.glob(osp.join(extensions_dir, '*.cpp'))
main_files = ['csrc/version.cpp', 'csrc/metis.cpp'] main_files = ['csrc/version.cpp', 'csrc/metis.cpp']
main_files = ['csrc/version.cpp']
for main, suffix in product(main_files, suffices): for main, suffix in product(main_files, suffices):
define_macros = [] define_macros = []
......
torch_sparse/__init__.py
View file @ 4833b184
...@@ -8,17 +8,8 @@ __version__ = '0.6.10' ...@@ -8,17 +8,8 @@ __version__ = '0.6.10'
suffix = 'cuda' if torch.cuda.is_available() else 'cpu' suffix = 'cuda' if torch.cuda.is_available() else 'cpu'
for library in [ for library in [
'_version', '_version', '_convert', '_diag', '_spmm', '_spspmm', '_metis', '_rw',
# '_convert', '_saint', '_sample', '_ego_sample', '_relabel'
# '_diag',
# '_spmm',
# '_spspmm',
'_metis',
# '_rw',
# '_saint',
# '_sample',
# '_ego_sample',
# '_relabel',
]: ]:
torch.ops.load_library(importlib.machinery.PathFinder().find_spec( torch.ops.load_library(importlib.machinery.PathFinder().find_spec(
f'{library}_{suffix}', [osp.dirname(__file__)]).origin) f'{library}_{suffix}', [osp.dirname(__file__)]).origin)
......
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