bug fix

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);
   }