pytorch 1.0.0 update

Author: rusty1s

cpu/dim_apply.h

@@ -1,6 +1,6 @@
 #pragma once
 
-#include <torch/torch.h>
+#include <torch/extension.h>
 
 #define DIM_APPLY3(TYPE1, TENSOR1, TYPE2, TENSOR2, TYPE3, TENSOR3, DIM, CODE)  \
   [&] {                                                                        \
@@ -17,7 +17,7 @@
     auto TENSOR3##_stride = TENSOR3.stride(DIM);                               \
                                                                                \
     auto dims = TENSOR1.dim();                                                 \
-    auto zeros = at::zeros(dims, torch::CPU(at::kLong));                       \
+    auto zeros = at::zeros(dims, TENSOR1.options().dtype(at::kLong));          \
     auto counter = zeros.data<int64_t>();                                      \
     bool has_finished = false;                                                 \
                                                                                \
@@ -76,7 +76,7 @@
     auto TENSOR4##_stride = TENSOR4.stride(DIM);                               \
                                                                                \
     auto dims = TENSOR1.dim();                                                 \
-    auto zeros = at::zeros(dims, torch::CPU(at::kLong));                       \
+    auto zeros = at::zeros(dims, TENSOR1.options().dtype(at::kLong));          \
     auto counter = zeros.data<int64_t>();                                      \
     bool has_finished = false;                                                 \
                                                                                \

cpu/scatter.cpp

@@ -1,4 +1,4 @@
-#include <torch/torch.h>
+#include <torch/extension.h>
 
 #include "dim_apply.h"
 

setup.py

@@ -15,7 +15,7 @@
                       ['cuda/scatter.cpp', 'cuda/scatter_kernel.cu'])
     ]
 
-__version__ = '1.0.4'
+__version__ = '1.0.5'
 url = 'https://github.com/rusty1s/pytorch_scatter'
 
 install_requires = []

torch_scatter/__init__.py

@@ -7,7 +7,7 @@
 from .max import scatter_max
 from .min import scatter_min
 
-__version__ = '1.0.4'
+__version__ = '1.0.5'
 
 __all__ = [
     'scatter_add',