pytorch 1.9.0

script/gcc.sh

-#!/bin/bash
-
-if [ "${TRAVIS_OS_NAME}" = "linux" ]; then
-  sudo add-apt-repository ppa:ubuntu-toolchain-r/test --yes
-  sudo apt update
-  sudo apt install gcc-7 g++-7 --yes
-  sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-7 60 \
-                           --slave /usr/bin/g++ g++ /usr/bin/g++-7
-  sudo update-alternatives --config gcc
-  gcc --version
-  g++ --version
-fi
-

script/install.sh

-#!/bin/bash
-
-if [ "${TRAVIS_OS_NAME}" != "windows" ]; then
-  python setup.py install
-else
-  echo "python setup.py install"
-  while sleep 30; do echo "=====[ Still running after $SECONDS seconds ]====="; done &
-  python setup.py install | tail -n 1000
-  kill %1
-fi

script/metis.sh

-#!/bin/bash
-
-METIS=metis-5.1.0
-export WITH_METIS=1
-
-wget -nv http://glaros.dtc.umn.edu/gkhome/fetch/sw/metis/${METIS}.tar.gz
-tar -xvzf ${METIS}.tar.gz
-cd ${METIS} || exit
-sed -i.bak -e 's/IDXTYPEWIDTH 32/IDXTYPEWIDTH 64/g' include/metis.h
-
-if [ "${TRAVIS_OS_NAME}" != "windows" ]; then
-  make config
-  make
-  sudo make install
-else
-  # Fix GKlib on Windows: https://github.com/jlblancoc/suitesparse-metis-for-windows/issues/6
-  sed -i.bak -e '61,69d' GKlib/gk_arch.h
-
-  cd build || exit
-
-  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/2017/BuildTools/VC/Tools/MSVC/14.16.27023/lib/x64
-  cp ../include/metis.h /c/Program\ Files\ \(x86\)/Microsoft\ Visual\ Studio/2017/BuildTools/VC/Tools/MSVC/14.16.27023/include
-
-  cd ..
-fi
-
-cd ..

script/torch.sh

-#!/bin/bash
-
-# https://github.com/pytorch/pytorch/commit/d2e16dd888a9b5fd55bd475d4fcffb70f388d4f0
-if [ "${TRAVIS_OS_NAME}" = "windows" ] && [ "${TORCH_VERSION}" = "1.7.0" ]; then
-  echo "Fix nvcc for PyTorch"
-  sed -i.bak -e 's/CONSTEXPR_EXCEPT_WIN_CUDA/const/g' /c/tools/miniconda3/envs/test/lib/site-packages/torch/include/torch/csrc/jit/api/module.h
-  sed -i.bak -e 's/return \*(this->value)/return \*((type\*)this->value)/g' /c/tools/miniconda3/envs/test/lib/site-packages/torch/include/pybind11/cast.h
-fi
-
-if [ "${TRAVIS_OS_NAME}" = "windows" ] && [ "${TORCH_VERSION}" = "1.7.0" ]; then
-  echo "Fix nvcc for PyTorch 1.7.0"
-  sed -i.bak '/static constexpr Symbol Kind/d' /c/tools/miniconda3/envs/test/lib/site-packages/torch/include/torch/csrc/jit/ir/ir.h
-fi

setup.py

@@ -48,7 +48,8 @@ def get_extensions():
         extra_link_args = ['-s']
 
         info = parallel_info()
-        if 'backend: OpenMP' in info and 'OpenMP not found' not in info:
+        if ('backend: OpenMP' in info and 'OpenMP not found' not in info
+                and sys.platform != 'darwin'):
             extra_compile_args['cxx'] += ['-DAT_PARALLEL_OPENMP']
             if sys.platform == 'win32':
                 extra_compile_args['cxx'] += ['/openmp']
@@ -101,7 +102,7 @@ tests_require = ['pytest', 'pytest-cov']
 
 setup(
     name='torch_sparse',
-    version='0.6.9',
+    version='0.6.10',
     author='Matthias Fey',
     author_email='matthias.fey@tu-dortmund.de',
     url='https://github.com/rusty1s/pytorch_sparse',

torch_sparse/__init__.py

@@ -3,7 +3,7 @@ import os.path as osp
 
 import torch
 
-__version__ = '0.6.9'
+__version__ = '0.6.10'
 
 suffix = 'cuda' if torch.cuda.is_available() else 'cpu'