Commit 3f14b88d authored by rusty1s's avatar rusty1s
Browse files

[ci skip]

parent aa55bf39
......@@ -11,7 +11,7 @@ jobs:
fail-fast: false
matrix:
# os: [ubuntu-16.04, macos-10.15, windows-latest]
os: [windows-latest]
os: [ubuntu-16.04, windows-latest]
python-version: [3.6] #, 3.7, 3.8, 3.9]
torch-version: [1.8.0] #, 1.9.0]
# cuda-version: ['cpu', 'cu101', 'cu102', 'cu111']
......
......@@ -13,7 +13,7 @@ sed -i.bak -e '61,69d' GKlib/gk_arch.h
cd build || exit
cmake .. # -A x64 # Ensure we are building with x64
cmake .. -A x64 # Ensure we are building with x64
cmake --build . --config "Release" --target ALL_BUILD
cp libmetis/Release/metis.lib /c/Program\ Files\ \(x86\)/Microsoft\ Visual\ Studio/2019/Enterprise/VC/Tools/MSVC/14.29.30037/lib/x64
cp ../include/metis.h /c/Program\ Files\ \(x86\)/Microsoft\ Visual\ Studio/2019/Enterprise/VC/Tools/MSVC/14.29.30037/include
......
......@@ -30,6 +30,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']
for main, suffix in product(main_files, suffices):
define_macros = []
......
......@@ -8,8 +8,17 @@ __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)
......
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