backward compatibility with torch-sparse==0.6.9

rusty1s · Jun 17, 2021 · a706309 · a706309
1 parent 48eed83
commit a706309
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/.github/workflows/building.yml b/.github/workflows/building.yml
@@ -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: |

diff --git a/csrc/metis.cpp b/csrc/metis.cpp
@@ -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);
   }
 }