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:
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: |
......
......@@ -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, nullptr, num_parts,
return partition_cpu(rowptr, col, optional_value, torch::nullopt, num_parts,
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