add free in test

inc/utils.h

@@ -5,4 +5,4 @@ typedef struct {
 } VecData;
 
 float vec_dist(VecData x, VecData y);
-VecData* fvecs_read(const char* filename, int* bounds);
+VecData* fvecs_read(const char* filename, int* bounds, int* num);

src/test.c

@@ -4,13 +4,18 @@
 
 
 int main() {
+    
+    int num = 0;
+    VecData* vecs = fvecs_read("../dataset/siftsmall_base.fvecs", NULL, &num);
+    printf("num: %d\n", num);
 
-    VecData* vecs = fvecs_read("../dataset/siftsmall_base.fvecs", NULL);
-    for (int i = 0; i < 10000; i++) {
-        printf("%f\n", vecs[i].vector[0]);
+
+
+    // Free memory
+    for (int i = 0; i < num; i++) {
+        free(vecs[i].vector);
     }
     free(vecs);
-
     return 0;
+}
 
-}

src/utils.c

@@ -14,7 +14,7 @@ float vec_dist(VecData x, VecData y) {
     return sqrt(sum);
 }
 
-VecData* fvecs_read(const char* filename, int* bounds) {
+VecData* fvecs_read(const char* filename, int* bounds, int* num) {
     FILE* fid = fopen(filename, "rb");
     if (fid == NULL) {
         fprintf(stderr, "I/O error : Unable to open the file %s\n", filename);
@@ -66,5 +66,8 @@ VecData* fvecs_read(const char* filename, int* bounds) {
 
 
     fclose(fid);
+    if (num != NULL) {
+        *num = n;
+    }
     return v;
 }