Fix memory leak when constructing a tensor from numpy (#98)
diff --git a/torch/csrc/generic/TensorMethods.cwrap b/torch/csrc/generic/TensorMethods.cwrap
index 49a46b2..b8f4a23 100644
--- a/torch/csrc/generic/TensorMethods.cwrap
+++ b/torch/csrc/generic/TensorMethods.cwrap
@@ -176,13 +176,9 @@
}
THTensor *result = THTensor_(newWithStorage)(storage, 0, sizes, strides);
- // newWithStorage increases refcount
- storage.release();
return result;
} else {
THTensor *result = THTensor_(newWithStorage)(storage, 0, NULL, NULL);
- // newWithStorage increases refcount
- storage.release();
return result;
}
}