reset

Author: rusty1s

.travis.yml

@@ -9,7 +9,7 @@ matrix:
 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.6 ]]; then pip install http://download.pytorch.org/whl/cpu/torch-0.4.0-cp36-cp36m-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
   - pip install codecov

cpu/dim_apply.h

@@ -17,7 +17,7 @@
     auto TENSOR3##_stride = TENSOR3.stride(DIM);                               \
                                                                                \
     auto dims = TENSOR1.dim();                                                 \
-    auto zeros = at::zeros(torch::CPU(at::kLong), {dims});                     \
+    auto zeros = at::zeros(dims, torch::CPU(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(torch::CPU(at::kLong), {dims});                     \
+    auto zeros = at::zeros(dims, torch::CPU(at::kLong));                       \
     auto counter = zeros.data<int64_t>();                                      \
     bool has_finished = false;                                                 \
                                                                                \