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
a7063092
Commit
a7063092
authored
Jun 17, 2021
by
rusty1s
Browse files
backward compatibility with torch-sparse==0.6.9
parent
48eed837
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
12 additions
and
2 deletions
+12
-2
.github/workflows/building.yml
.github/workflows/building.yml
+11
-1
csrc/metis.cpp
csrc/metis.cpp
+1
-1
No files found.
.github/workflows/building.yml
View file @
a7063092
...
...
@@ -49,10 +49,20 @@ jobs:
python -c "import torch; print('PyTorch:', torch.__version__)"
python -c "import torch; print('CUDA:', torch.version.cuda)"
-
name
:
Install
internal dependencies
-
name
:
Install
torch-scatter
run
:
|
pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-${{ matrix.torch-version }}+${{ matrix.cuda-version }}.html
-
name
:
Install METIS
if
:
${{ runner.os != 'Windows' }}
run
:
|
bash .github/workflows/metis.sh
-
name
:
Install METIS on Windows
if
:
${{ runner.os == 'Windows' }}
run
:
|
bash .github/workflows/metis-${{ runner.os }}.sh
-
name
:
Install main package for CPU
if
:
${{ matrix.cuda-version == 'cpu' }}
run
:
|
...
...
csrc/metis.cpp
View file @
a7063092
...
...
@@ -21,7 +21,7 @@ torch::Tensor partition(torch::Tensor rowptr, torch::Tensor col,
AT_ERROR
(
"Not compiled with CUDA support"
);
#endif
}
else
{
return
partition_cpu
(
rowptr
,
col
,
optional_value
,
nullpt
r
,
num_parts
,
return
partition_cpu
(
rowptr
,
col
,
optional_value
,
torch
::
null
o
pt
,
num_parts
,
recursive
);
}
}
...
...
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