fix

45d29d1a · rusty1s · 1ad47fb2 · 45d29d1a · 45d29d1a
Commit 45d29d1a authored Feb 23, 2020 by rusty1s
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 3 deletions

script/metis.sh script/metis.sh +2 -2

torch_sparse/__init__.py torch_sparse/__init__.py +2 -1

No files found.
--- a/script/metis.sh
+++ b/script/metis.sh
@@ -19,8 +19,8 @@ else

  cmake ..
  cmake --build . --config "Release" --target ALL_BUILD
-  cp libetmis/Release/metis.lib ../..
-  # cp libmetis/Release/metis.lib /c/Program\ Files\ \(x86\)/Microsoft\ Visual\ Studio/2017/BuildTools/VC/Tools/MSVC/14.16.27023/lib/x64
+  # cp libmetis/Release/metis.lib ../..
+  cp libmetis/Release/metis.lib /c/Program\ Files\ \(x86\)/Microsoft\ Visual\ Studio/2017/BuildTools/VC/Tools/MSVC/14.16.27023/lib/x64

  # cmake .. -DSHARED=1
  # cmake --build . --config "Release" --target ALL_BUILD

--- a/torch_sparse/__init__.py
+++ b/torch_sparse/__init__.py
@@ -8,7 +8,8 @@ expected_torch_version = (1, 4)

 try:
    for library in [
-            '_version', '_convert', '_diag', '_spmm', '_spspmm', '_metis'
+            '_version', '_convert', '_diag', '_spmm', '_spspmm',
+            '_metis_wrapper'
    ]:
        torch.ops.load_library(importlib.machinery.PathFinder().find_spec(
            library, [osp.dirname(__file__)]).origin)