From ccec96ba54f25c96d12808daaf0c1ee775483adf Mon Sep 17 00:00:00 2001
From: Andrew Kane <andrew@ankane.org>
Date: Thu, 17 Nov 2022 14:12:49 -0800
Subject: [PATCH] Fixed invalid memory alloc request size error - fixes #43

---
 CHANGELOG.md   | 4 ++++
 src/ivfutils.c | 2 +-
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/CHANGELOG.md b/CHANGELOG.md
index 691d499..9210464 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -1,3 +1,7 @@
+## 0.3.2 (unreleased)
+
+- Fixed `invalid memory alloc request size` error
+
 ## 0.3.1 (2022-11-02)
 
 If upgrading from 0.2.7 or 0.3.0, [recreate](https://github.com/pgvector/pgvector#031) all `ivfflat` indexes after upgrading to ensure all data is indexed.
diff --git a/src/ivfutils.c b/src/ivfutils.c
index 27ff2ac..3dc39c0 100644
--- a/src/ivfutils.c
+++ b/src/ivfutils.c
@@ -10,7 +10,7 @@
 VectorArray
 VectorArrayInit(int maxlen, int dimensions)
 {
-	VectorArray res = palloc0(VECTOR_ARRAY_SIZE(maxlen, dimensions));
+	VectorArray res = palloc_extended(VECTOR_ARRAY_SIZE(maxlen, dimensions), MCXT_ALLOC_ZERO | MCXT_ALLOC_HUGE);
 
 	res->length = 0;
 	res->maxlen = maxlen;