only cpu fix

Author: rusty1s

.travis.yml

@@ -3,6 +3,7 @@ sudo: required
 dist: trusty
 matrix:
   include:
+    - python: 2.7
     - python: 3.5
     - python: 3.6
 addons:
@@ -16,6 +17,7 @@ before_install:
   - export CC="gcc-4.9"
   - export CXX="g++-4.9"
 install:
+  - if [[ $TRAVIS_PYTHON_VERSION == 2.7 ]]; then pip install https://download.pytorch.org/whl/cpu/torch-1.0.0-cp27-cp27mu-linux_x86_64.whl; fi
   - if [[ $TRAVIS_PYTHON_VERSION == 3.5 ]]; then pip install https://download.pytorch.org/whl/cpu/torch-1.0.0-cp35-cp35m-linux_x86_64.whl; fi
   - if [[ $TRAVIS_PYTHON_VERSION == 3.6 ]]; then pip install https://download.pytorch.org/whl/cpu/torch-1.0.0-cp36-cp36m-linux_x86_64.whl; fi
   - pip install pycodestyle

setup.py

@@ -3,14 +3,8 @@
 import torch
 from torch.utils.cpp_extension import CppExtension, CUDAExtension, CUDA_HOME
 
-__version__ = '0.3.0'
-url = 'https://github.com/rusty1s/pytorch_sparse'
-
-install_requires = ['scipy']
-setup_requires = ['pytest-runner']
-tests_require = ['pytest', 'pytest-cov']
 ext_modules = [CppExtension('torch_sparse.spspmm_cpu', ['cpu/spspmm.cpp'])]
-cmdclass = {}
+cmdclass = {'build_ext': torch.utils.cpp_extension.BuildExtension}
 
 if CUDA_HOME is not None:
     if platform.system() == 'Windows':
@@ -26,7 +20,13 @@
         CUDAExtension('torch_sparse.unique_cuda',
                       ['cuda/unique.cpp', 'cuda/unique_kernel.cu']),
     ]
-    cmdclass['build_ext'] = torch.utils.cpp_extension.BuildExtension
+
+__version__ = '0.3.0'
+url = 'https://github.com/rusty1s/pytorch_sparse'
+
+install_requires = ['scipy']
+setup_requires = ['pytest-runner']
+tests_require = ['pytest', 'pytest-cov']
 
 setup(
     name='torch_sparse',