Commit a7063092 authored by rusty1s's avatar rusty1s
Browse files

backward compatibility with torch-sparse==0.6.9

parent 48eed837
...@@ -49,10 +49,20 @@ jobs: ...@@ -49,10 +49,20 @@ jobs:
python -c "import torch; print('PyTorch:', torch.__version__)" python -c "import torch; print('PyTorch:', torch.__version__)"
python -c "import torch; print('CUDA:', torch.version.cuda)" python -c "import torch; print('CUDA:', torch.version.cuda)"
- name: Install internal dependencies - name: Install torch-scatter
run: | run: |
pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-${{ matrix.torch-version }}+${{ matrix.cuda-version }}.html 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 - name: Install main package for CPU
if: ${{ matrix.cuda-version == 'cpu' }} if: ${{ matrix.cuda-version == 'cpu' }}
run: | run: |
......
...@@ -21,7 +21,7 @@ torch::Tensor partition(torch::Tensor rowptr, torch::Tensor col, ...@@ -21,7 +21,7 @@ torch::Tensor partition(torch::Tensor rowptr, torch::Tensor col,
AT_ERROR("Not compiled with CUDA support"); AT_ERROR("Not compiled with CUDA support");
#endif #endif
} else { } else {
return partition_cpu(rowptr, col, optional_value, nullptr, num_parts, return partition_cpu(rowptr, col, optional_value, torch::nullopt, num_parts,
recursive); recursive);
} }
} }
......
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