travis gcc+4.9

.travis.yml

@@ -6,6 +6,16 @@ matrix:
     - python: 2.7
     - python: 3.5
     - python: 3.6
+addons:
+  apt:
+    sources:
+      - ubuntu-toolchain-r-test
+    packages:
+      - gcc-4.9
+      - g++-4.9
+before_install:
+  - export CC="gcc-4.9"
+  - export CXX="g++-4.9"
 install:
   - if [[ $TRAVIS_PYTHON_VERSION == 2.7 ]]; then pip install http://download.pytorch.org/whl/cpu/torch-0.4.1-cp27-cp27mu-linux_x86_64.whl; fi
   - if [[ $TRAVIS_PYTHON_VERSION == 3.5 ]]; then pip install http://download.pytorch.org/whl/cpu/torch-0.4.1-cp35-cp35m-linux_x86_64.whl; fi

setup.py

@@ -14,10 +14,10 @@ cmdclass = {}
 if torch.cuda.is_available():
     ext_modules += [
         CUDAExtension(
-            'spspmm_cuda', ['cuda/spspmm.cpp', 'cuda/spspmm_kernel.cu'],
-            extra_compile_args={
-                'nvcc': ['-lcusparse']
-            })
+            'spspmm_cuda',
+            ['cuda/spspmm.cpp', 'cuda/spspmm_kernel.cu'],
+            extra_link_args=['-l', 'cusparse'],
+        )
     ]
     cmdclass['build_ext'] = BuildExtension