Merge branch 'master' of github.com:rusty1s/pytorch_scatter

Author: rusty1s

.travis.yml

@@ -18,7 +18,7 @@ before_install:
   - 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
+  - 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
   - if [[ $TRAVIS_PYTHON_VERSION == 3.6 ]]; then pip install http://download.pytorch.org/whl/cpu/torch-0.4.1-cp36-cp36m-linux_x86_64.whl; fi
   - pip install pycodestyle
   - pip install flake8

README.md

@@ -35,7 +35,7 @@ All included operations work on varying data types, are implemented both for CPU
 
 ## Installation
 
-Ensure that at least PyTorch 0.4.1 is installed and verify that `cuda/bin` and `cuda/install` are in your `$PATH` and `$CPATH` respectively, *e.g.*:
+Ensure that at least PyTorch 0.4.1 is installed and verify that `cuda/bin` and `cuda/include` are in your `$PATH` and `$CPATH` respectively, *e.g.*:
 
 ```
 $ python -c "import torch; print(torch.__version__)"
@@ -45,7 +45,7 @@ $ echo $PATH
 >>> /usr/local/cuda/bin:...
 
 $ echo $CPATH
->>> /usr/local/cuda/install:...
+>>> /usr/local/cuda/include:...
 ```
 
 Then run:

cuda/atomics.cuh

@@ -1,3 +1,5 @@
+#pragma once
+
 #define ATOMIC(NAME)                                                           \
   template <typename scalar, size_t size> struct Atomic##NAME##IntegerImpl;    \
                                                                                \