bug fix

9a4f0247 · Martin Reinecke · 32cb03a0 · 9a4f0247
Commit 9a4f0247 authored Apr 27, 2019 by Martin Reinecke
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

pocketfft.cc pocketfft.cc +2 -2

No files found.
--- a/pocketfft.cc
+++ b/pocketfft.cc
@@ -2079,7 +2079,7 @@ template<typename T> arr<char> alloc_tmp(const shape_t &shape,
  for (size_t i=0; i<ndim; ++i)
    fullsize*=shape[i];
  auto othersize = fullsize/axsize;
-  auto tmpsize = axsize*((othersize>VTYPE<T>::vlen) ? VTYPE<T>::vlen : 1);
+  auto tmpsize = axsize*((othersize>=VTYPE<T>::vlen) ? VTYPE<T>::vlen : 1);
  return arr<char>(tmpsize*elemsize);
  }
 template<typename T> arr<char> alloc_tmp(const shape_t &shape,
@@ -2094,7 +2094,7 @@ template<typename T> arr<char> alloc_tmp(const shape_t &shape,
    {
    auto axsize = shape[axes[i]];
    auto othersize = fullsize/axsize;
-    tmpsize = max(tmpsize, axsize*((othersize>VTYPE<T>::vlen) ? VTYPE<T>::vlen : 1));
+    tmpsize = max(tmpsize, axsize*((othersize>=VTYPE<T>::vlen) ? VTYPE<T>::vlen : 1));
    }
  return arr<char>(tmpsize*elemsize);
  }