version update

Author: rusty1s

README.md

@@ -23,14 +23,27 @@ All included operations work on varying data types and are implemented both for
 
 ## Installation
 
-Check that `nvcc` is accessible from terminal, e.g. `nvcc --version`.
-If not, add cuda (`/usr/local/cuda/bin`) to your `$PATH`.
+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.*:
+
+```
+$ python -c "import torch; print(torch.__version__)"
+>>> 0.4.1
+
+$ echo $PATH
+>>> /usr/local/cuda/bin:...
+
+$ echo $CPATH
+>>> /usr/local/cuda/install:...
+```
+
 Then run:
 
 ```
 pip install cffi torch-cluster
 ```
 
+If you are running into any installation problems, please create an [issue](https://github.com/rusty1s/pytorch_cluster/issues).
+
 ## Graclus
 
 A greedy clustering algorithm of picking an unmarked vertex and matching it with one its unmarked neighbors (that maximizes its edge weight).

aten/TH/generic/THGrid.c

@@ -2,24 +2,27 @@
 #define TH_GENERIC_FILE "generic/THGrid.c"
 #else
 
-void THTensor_(grid)(THLongTensor *self, THTensor *pos, THTensor *size, THLongTensor *count) {
+void THTensor_(grid)(THLongTensor *self, THTensor *pos, THTensor *size,
+                     THLongTensor *count) {
   int64_t *selfData = THLongTensor_data(self);
   real *posData = THTensor_(data)(pos);
   real *sizeData = THTensor_(data)(size);
   int64_t posStride0 = THTensor_(stride)(pos, 0);
   int64_t posStride1 = THTensor_(stride)(pos, 1);
   int64_t *countData = THLongTensor_data(count);
 
-  ptrdiff_t n, d; int64_t coef, value;
+  ptrdiff_t n, d;
+  int64_t coef, value;
   for (n = 0; n < THTensor_(size)(pos, 0); n++) {
-    coef = 1; value = 0;
+    coef = 1;
+    value = 0;
     for (d = 0; d < THTensor_(size)(pos, 1); d++) {
-      value += coef * (int64_t) (posData[d * posStride1] / sizeData[d]);
+      value += coef * (int64_t)(posData[d * posStride1] / sizeData[d]);
       coef *= countData[d];
     }
     posData += posStride0;
     selfData[n] = value;
   }
 }
 
-#endif  // TH_GENERIC_FILE
+#endif // TH_GENERIC_FILE

setup.py

@@ -2,7 +2,7 @@
 
 from setuptools import setup, find_packages
 
-__version__ = '1.1.2'
+__version__ = '1.1.3'
 url = 'https://github.com/rusty1s/pytorch_cluster'
 
 install_requires = ['cffi']

torch_cluster/__init__.py

@@ -1,6 +1,6 @@
 from .graclus import graclus_cluster
 from .grid import grid_cluster
 
-__version__ = '1.1.2'
+__version__ = '1.1.3'
 
 __all__ = ['graclus_cluster', 'grid_cluster', '__version__']