fix external allocation

AMYPAD · Jan 21, 2021 · 9ac352d · 9ac352d
1 parent 9875938
commit 9ac352d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cuvec/include/pycuvec.cuh b/cuvec/include/pycuvec.cuh
@@ -204,7 +204,7 @@ template <class T> struct PyCuVec_tp {
 template <class T> PyCuVec<T> *PyCuVec_new() {
   static PyCuVec_tp<T> Vector_T;
   if (PyType_Ready(&Vector_T.tp_obj) < 0) return NULL;
-  return (PyCuVec<T> *)Vector_T.tp_obj.tp_alloc(&Vector_T.tp_obj, 0);
+  return (PyCuVec<T> *)Vector_T.tp_obj.tp_alloc(&Vector_T.tp_obj, 1);
 }
 template <class T> PyCuVec<T> *PyCuVec_zeros(std::vector<Py_ssize_t> shape) {
   PyCuVec<T> *self = PyCuVec_new<T>();