xmem bugs

source/tensor/Main.cpp

@@ -44,6 +44,9 @@ int main( int argc, const char ** argv )
 {
 {
     //_CrtSetBreakAlloc(123);
     //_CrtSetBreakAlloc(123);
 
 
+    TestXMem();
+    return 0;
+
     /* a tiny test */
     /* a tiny test */
     //SmallTest();
     //SmallTest();
 
 

source/tensor/XMem.cpp

@@ -279,7 +279,6 @@ initialize the index
 */
 */
 void XMem::SetIndex(INT_64 indexSize, MTYPE minSizeFirst, int minSizeNum)
 void XMem::SetIndex(INT_64 indexSize, MTYPE minSizeFirst, int minSizeNum)
 {
 {
-
 	delete freeMemIndex;
 	delete freeMemIndex;
 	delete minSizeIndex;
 	delete minSizeIndex;
 
 

source/tensor/XMem.h

@@ -53,22 +53,6 @@ typedef long long          INT_64;
 #define MIN_BLOCK_SIZE_FOR_MEMPOOL 128 * 1024 * 1024
 #define MIN_BLOCK_SIZE_FOR_MEMPOOL 128 * 1024 * 1024
 #define MIN_BLOCK_NUM_FOR_MEMPOOL 1024
 #define MIN_BLOCK_NUM_FOR_MEMPOOL 1024
 
 
-/* memory block */
-struct XMemBlock
-{
-    /* pointer to where to start */
-    void * mem;
-
-    /* size of the block */
-    MTYPE size;
-
-    /* size of the used memory in this block */
-    MTYPE used;
-
-    /* disired size of the block */
-    MTYPE sizeDesired;
-};
-
 /* 
 /* 
 mode of runnig a memory pool 
 mode of runnig a memory pool 
 - UNI_FREE: free all memory space when the memory allocation is no use
 - UNI_FREE: free all memory space when the memory allocation is no use
@@ -112,6 +96,25 @@ struct MPieceNode
 
 
     /* pointer to the head of a memory piece */
     /* pointer to the head of a memory piece */
     void * p;
     void * p;
+
+    /* header of the memory piece */
+    MHeader * head;
+};
+
+/* memory block */
+struct XMemBlock
+{
+    /* pointer to where to start */
+    void * mem;
+
+    /* size of the block */
+    MTYPE size;
+
+    /* size of the used memory in this block */
+    MTYPE used;
+
+    /* desired size of the block */
+    MTYPE sizeDesired;
 };
 };
 
 
 /* 
 /* 

source/tensor/test/TXMem.cpp

@@ -33,50 +33,65 @@ bool TestXMemCase1()
     int blcokSize = 16;
     int blcokSize = 16;
     int testNum = caseNum * 10;
     int testNum = caseNum * 10;
 
 
-    for(int i = 0, scalar = 1; i < 3; i++){
-        XMem mem;
-        mem.Initialize(-1, FREE_ON_THE_FLY, blcokSize * sizeof(int) * scalar * scalar, 1000, 0);
-        mem.SetIndex(10000, blcokSize * sizeof(int) / 2);
+    int devIDs[2];
+    int devNum = 1;
+    devIDs[0] = -1;
 
 
-        srand(907);
+    /*if (GDevs.nGPU > 0) {
+        devIDs[1] = 0;
+        devNum = 2;
+        devIDs[0] = 0;
+        devNum = 1;
+    }*/
 
 
-        int ** p = new int*[caseNum];
-        int * size = new int[caseNum];
 
 
-        for(int i = 0; i < caseNum; i++){
-            p[i] = NULL;
-            size[i] = rand() % (2*blcokSize);
-        }
+    for (int id = 0; id < devNum; id++) {
+        int devID = devIDs[id];
+        for (int i = 0, scalar = 1; i < 3; i++) {
+            XMem mem;
+            mem.Initialize(devID, FREE_ON_THE_FLY, blcokSize * sizeof(int) * scalar * scalar, 1000, 0);
+            mem.SetIndex(10000, blcokSize * sizeof(int) / 2);
 
 
-        for(int i = 0; i < testNum * scalar; i++){
-            int j = rand() % caseNum;
+            srand(907);
 
 
-            if(p[j] == NULL){
-                p[j] = (int*)mem.AllocStandard(mem.devID, size[j] * sizeof(int));
-                for(int k = 0; k < size[j]; k++)
-                    p[j][k] = j;
-            }
-            else{
-                mem.ReleaseStandard(mem.devID, p[j]);
-                for(int k = 0; k < size[j]; k++)
-                    p[j][k] = -1;
-                p[j] = NULL;
+            int ** p = new int*[caseNum];
+            int * size = new int[caseNum];
+
+            for (int i = 0; i < caseNum; i++) {
+                p[i] = NULL;
+                size[i] = rand() % (2 * blcokSize);
             }
             }
 
 
-            for(int k = 0; k < caseNum; k++){
-                if(p[k] != NULL){
-                    for(int o = 0; o < size[k]; o++){
-                        if(p[k][o] != k){
-                            ok = false;
+            for (int i = 0; i < testNum * scalar; i++) {
+                int j = rand() % caseNum;
+
+                if (p[j] == NULL) {
+                    p[j] = (int*)mem.AllocStandard(mem.devID, size[j] * sizeof(int));
+                    for (int k = 0; k < size[j]; k++)
+                        p[j][k] = j;
+                }
+                else {
+                    mem.ReleaseStandard(mem.devID, p[j]);
+                    for (int k = 0; k < size[j]; k++)
+                        p[j][k] = -1;
+                    p[j] = NULL;
+                }
+
+                for (int k = 0; k < caseNum; k++) {
+                    if (p[k] != NULL) {
+                        for (int o = 0; o < size[k]; o++) {
+                            if (p[k][o] != k) {
+                                ok = false;
+                            }
                         }
                         }
                     }
                     }
                 }
                 }
             }
             }
-        }
 
 
-        delete[] p;
-        delete[] size;
-        scalar *= 2;
+            delete[] p;
+            delete[] size;
+            scalar *= 2;
+        }
     }
     }
 
 
     return ok;
     return ok;