Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
torch-sparse
Commits
3f14b88d
Commit
3f14b88d
authored
Jun 26, 2021
by
rusty1s
Browse files
[ci skip]
parent
aa55bf39
Changes
4
Hide whitespace changes
Inline
Side-by-side
Showing
4 changed files
with
14 additions
and
4 deletions
+14
-4
.github/workflows/building-conda.yml
.github/workflows/building-conda.yml
+1
-1
.github/workflows/metis-Windows.sh
.github/workflows/metis-Windows.sh
+1
-1
setup.py
setup.py
+1
-0
torch_sparse/__init__.py
torch_sparse/__init__.py
+11
-2
No files found.
.github/workflows/building-conda.yml
View file @
3f14b88d
...
...
@@ -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']
...
...
.github/workflows/metis-Windows.sh
View file @
3f14b88d
...
...
@@ -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
...
...
setup.py
View file @
3f14b88d
...
...
@@ -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
=
[]
...
...
torch_sparse/__init__.py
View file @
3f14b88d
...
...
@@ -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
)
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment