version up

rusty1s · Jul 27, 2019 · 0fb1f1d · 0fb1f1d
1 parent 4fa1fb2
commit 0fb1f1d
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 5 deletions.
diff --git a/cpu/sampler.cpp b/cpu/sampler.cpp
@@ -1,9 +1,7 @@
-#include <ATen/CPUGenerator.h>
 #include <torch/extension.h>
 
 at::Tensor neighbor_sampler(at::Tensor start, at::Tensor cumdeg, size_t size,
                             float factor) {
-  at::CPUGenerator *generator = at::detail::getDefaultCPUGenerator();
 
   auto start_ptr = start.data<int64_t>();
   auto cumdeg_ptr = cumdeg.data<int64_t>();
@@ -24,7 +22,7 @@ at::Tensor neighbor_sampler(at::Tensor start, at::Tensor cumdeg, size_t size,
     std::unordered_set<int64_t> set;
     if (size_i < 0.7 * float(num_neighbors)) {
       while (set.size() < size_i) {
-        int64_t z = generator->random() % num_neighbors;
+        int64_t z = rand() % num_neighbors;
         set.insert(z + low);
       }
       std::vector<int64_t> v(set.begin(), set.end());

diff --git a/setup.py b/setup.py
@@ -29,7 +29,7 @@
                       ['cuda/rw.cpp', 'cuda/rw_kernel.cu']),
     ]
 
-__version__ = '1.4.3a1'
+__version__ = '1.4.3'
 url = 'https://github.com/rusty1s/pytorch_cluster'
 
 install_requires = ['scipy']

diff --git a/torch_cluster/__init__.py b/torch_cluster/__init__.py
@@ -7,7 +7,7 @@
 from .rw import random_walk
 from .sampler import neighbor_sampler
 
-__version__ = '1.4.3a1'
+__version__ = '1.4.3'
 
 __all__ = [
     'graclus_cluster',