Skip to content

GitLab

Commit 45d29d1a authored by rusty1s's avatar rusty1s
Browse files

fix

parent 1ad47fb2
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 3 deletions
script/metis.sh
View file @ 45d29d1a
...@@ -19,8 +19,8 @@ else ...@@ -19,8 +19,8 @@ else
cmake .. cmake ..
cmake --build . --config "Release" --target ALL_BUILD cmake --build . --config "Release" --target ALL_BUILD
cp libetmis/Release/metis.lib ../.. # 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 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 .. -DSHARED=1
# cmake --build . --config "Release" --target ALL_BUILD # cmake --build . --config "Release" --target ALL_BUILD
......
torch_sparse/__init__.py
View file @ 45d29d1a
...@@ -8,7 +8,8 @@ expected_torch_version = (1, 4) ...@@ -8,7 +8,8 @@ expected_torch_version = (1, 4)
try: try:
for library in [ 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( torch.ops.load_library(importlib.machinery.PathFinder().find_spec(
library, [osp.dirname(__file__)]).origin) library, [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