Fix hugepages memory allocation issues.

When mmap() fails, it returns MAP_FAILED.
If the mmap() with hugepages fails, call mmap() again without hugepages.

Signed-off-by: ShinYee <[email protected]>

Author: xman

SLIDE/Layer.h

@@ -55,7 +55,12 @@ class Layer
         void* ptr = mmap(NULL, size,
             PROT_READ | PROT_EXEC | PROT_WRITE, MAP_PRIVATE | MAP_ANONYMOUS | MAP_HUGETLB,
             -1, 0);
-        if (ptr == NULL)
+        if (ptr == MAP_FAILED){
+            ptr = mmap(NULL, size,
+                PROT_READ | PROT_EXEC | PROT_WRITE, MAP_PRIVATE | MAP_ANONYMOUS,
+                -1, 0);
+        }
+        if (ptr == MAP_FAILED)
             std::cout << "mmap fail! No new layer!" << std::endl;
         return ptr;};
     void operator delete(void * pointer){munmap(pointer, sizeof(Layer));};

SLIDE/Network.h

@@ -28,9 +28,19 @@ class Network
 	~Network();
 	void * operator new(size_t size){
 	    cout << "new Network" << endl;
-	    return mmap(NULL, size,
-            PROT_READ | PROT_EXEC | PROT_WRITE, MAP_PRIVATE | MAP_ANONYMOUS | MAP_HUGETLB,
-            -1, 0);};
+	    void* ptr = mmap(NULL, size,
+	        PROT_READ | PROT_EXEC | PROT_WRITE, MAP_PRIVATE | MAP_ANONYMOUS | MAP_HUGETLB,
+	        -1, 0);
+	    if (ptr == MAP_FAILED){
+	        ptr = mmap(NULL, size,
+	            PROT_READ | PROT_EXEC | PROT_WRITE, MAP_PRIVATE | MAP_ANONYMOUS,
+	            -1, 0);
+	    }
+	    if (ptr == MAP_FAILED){
+	        std::cout << "mmap failed at Network." << std::endl;
+	    }
+	    return ptr;
+	}
 	void operator delete(void * pointer){munmap(pointer, sizeof(Network));};
 };
 

SLIDE/Node.h

@@ -19,18 +19,35 @@ struct train {
     int _ActiveinputIds;
 
     void * operator new(size_t size){
-        return mmap(NULL, size,
+        void* ptr = mmap(NULL, size,
             PROT_READ | PROT_EXEC | PROT_WRITE, MAP_PRIVATE | MAP_ANONYMOUS | MAP_HUGETLB,
-            -1, 0);};
+            -1, 0);
+        if (ptr == MAP_FAILED){
+            ptr = mmap(NULL, size,
+                PROT_READ | PROT_EXEC | PROT_WRITE, MAP_PRIVATE | MAP_ANONYMOUS,
+                -1, 0);
+        }
+        if (ptr == MAP_FAILED){
+            std::cout << "mmap failed at train." << std::endl;
+        }
+        return ptr;
+    }
     void* operator new (std::size_t size, const std::nothrow_t& nothrow_value){return operator new (size);};
     void* operator new (std::size_t size, void* ptr){return operator new (size);};
     void* operator new[] (std::size_t size){
         void* ptr = mmap(NULL, size,
             PROT_READ | PROT_EXEC | PROT_WRITE, MAP_PRIVATE | MAP_ANONYMOUS | MAP_HUGETLB | MAP_HUGE_1GB,
             -1, 0);
-        if (ptr == NULL)
+        if (ptr == MAP_FAILED){
+            ptr = mmap(NULL, size,
+                PROT_READ | PROT_EXEC | PROT_WRITE, MAP_PRIVATE | MAP_ANONYMOUS,
+                -1, 0);
+        }
+        if (ptr == MAP_FAILED){
             std::cout << "mmap fail! No train array!" << std::endl;
-        return ptr;};
+        }
+        return ptr;
+    }
     void* operator new[] (std::size_t size, const std::nothrow_t& nothrow_value){return operator new (size);};
     void* operator new[] (std::size_t size, void* ptr){return operator new (size);};
 
@@ -86,16 +103,36 @@ class Node
 
     void * operator new(size_t size){
         std::cout << "new Node" << std::endl;
-        return mmap(NULL, size,
+        void* ptr = mmap(NULL, size,
             PROT_READ | PROT_EXEC | PROT_WRITE, MAP_PRIVATE | MAP_ANONYMOUS | MAP_HUGETLB | MAP_HUGE_1GB,
-            -1, 0);};
+            -1, 0);
+        if (ptr == MAP_FAILED){
+            ptr = mmap(NULL, size,
+                PROT_READ | PROT_EXEC | PROT_WRITE, MAP_PRIVATE | MAP_ANONYMOUS,
+                -1, 0);
+        }
+        if (ptr == MAP_FAILED){
+            std::cout << "mmap failed at Node." << std::endl;
+        }
+        return ptr;
+    }
     void* operator new (std::size_t size, const std::nothrow_t& nothrow_value){return operator new (size);};
     void* operator new (std::size_t size, void* ptr){return operator new (size);};
     void* operator new[] (std::size_t size){
         std::cout << "new Node array" << std::endl;
-        return mmap(NULL, size,
+        void* ptr = mmap(NULL, size,
             PROT_READ | PROT_EXEC | PROT_WRITE, MAP_PRIVATE | MAP_ANONYMOUS | MAP_HUGETLB | MAP_HUGE_1GB,
-            -1, 0);};
+            -1, 0);
+        if (ptr == MAP_FAILED){
+            ptr = mmap(NULL, size,
+                PROT_READ | PROT_EXEC | PROT_WRITE, MAP_PRIVATE | MAP_ANONYMOUS,
+                -1, 0);
+        }
+        if (ptr == MAP_FAILED){
+            std::cout << "mmap failed at Node array." << std::endl;
+        }
+        return ptr;
+    }
     void* operator new[] (std::size_t size, const std::nothrow_t& nothrow_value){return operator new (size);};
     void* operator new[] (std::size_t size, void* ptr){return operator new (size);};