Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
Fix compilation for systems without MAP_HUGETLB.
authorTom Cornebize <tom.cornebize@ensimag.grenoble-inp.fr>
Thu, 18 May 2017 17:00:39 +0000 (19:00 +0200)
committerTom Cornebize <tom.cornebize@ensimag.grenoble-inp.fr>
Thu, 18 May 2017 17:00:39 +0000 (19:00 +0200)
src/smpi/smpi_shared.cpp

index ee0687d..32bdecd 100644 (file)
@@ -225,9 +225,13 @@ void* smpi_shared_malloc_partial(size_t size, size_t* shared_block_offsets, int
 {
   char *huge_page_mount_point = xbt_cfg_get_string("smpi/shared-malloc-hugepage");
   const bool use_huge_page = huge_page_mount_point[0] != '\0';
 {
   char *huge_page_mount_point = xbt_cfg_get_string("smpi/shared-malloc-hugepage");
   const bool use_huge_page = huge_page_mount_point[0] != '\0';
+#ifndef MAP_HUGETLB
+    xbt_assert(!use_huge_page, "Huge pages are not available on your system.");
+#endif
   smpi_shared_malloc_blocksize = static_cast<unsigned long>(xbt_cfg_get_double("smpi/shared-malloc-blocksize"));
   void *mem, *allocated_ptr;
   size_t allocated_size;
   smpi_shared_malloc_blocksize = static_cast<unsigned long>(xbt_cfg_get_double("smpi/shared-malloc-blocksize"));
   void *mem, *allocated_ptr;
   size_t allocated_size;
+#ifdef MAP_HUGETLB
   if(use_huge_page) {
     xbt_assert(smpi_shared_malloc_blocksize == HUGE_PAGE_SIZE, "the block size of shared malloc should be equal to the size of a huge page.");
     allocated_size = size + 2*smpi_shared_malloc_blocksize;
   if(use_huge_page) {
     xbt_assert(smpi_shared_malloc_blocksize == HUGE_PAGE_SIZE, "the block size of shared malloc should be equal to the size of a huge page.");
     allocated_size = size + 2*smpi_shared_malloc_blocksize;
@@ -236,6 +240,10 @@ void* smpi_shared_malloc_partial(size_t size, size_t* shared_block_offsets, int
     xbt_assert(smpi_shared_malloc_blocksize % PAGE_SIZE == 0, "the block size of shared malloc should be a multiple of the page size.");
     allocated_size = size;
   }
     xbt_assert(smpi_shared_malloc_blocksize % PAGE_SIZE == 0, "the block size of shared malloc should be a multiple of the page size.");
     allocated_size = size;
   }
+#else
+  xbt_assert(smpi_shared_malloc_blocksize % PAGE_SIZE == 0, "the block size of shared malloc should be a multiple of the page size.");
+  allocated_size = size;
+#endif
 
 
   /* First reserve memory area */
 
 
   /* First reserve memory area */
@@ -245,16 +253,20 @@ void* smpi_shared_malloc_partial(size_t size, size_t* shared_block_offsets, int
   xbt_assert(allocated_ptr != MAP_FAILED, "Failed to allocate %zuMiB of memory. Run \"sysctl vm.overcommit_memory=1\" as root "
                                 "to allow big allocations.\n",
              size >> 20);
   xbt_assert(allocated_ptr != MAP_FAILED, "Failed to allocate %zuMiB of memory. Run \"sysctl vm.overcommit_memory=1\" as root "
                                 "to allow big allocations.\n",
              size >> 20);
-
+#ifdef MAP_HUGETLB
   if(use_huge_page)
     mem = (void*)ALIGN_UP((uint64_t)allocated_ptr, HUGE_PAGE_SIZE);
   else
     mem = allocated_ptr;
   if(use_huge_page)
     mem = (void*)ALIGN_UP((uint64_t)allocated_ptr, HUGE_PAGE_SIZE);
   else
     mem = allocated_ptr;
+#else
+  mem = allocated_ptr;
+#endif
 
   XBT_DEBUG("global shared allocation. Blocksize %lu", smpi_shared_malloc_blocksize);
   /* Create a fd to a new file on disk, make it smpi_shared_malloc_blocksize big, and unlink it.
    * It still exists in memory but not in the file system (thus it cannot be leaked). */
   /* Create bogus file if not done already */
 
   XBT_DEBUG("global shared allocation. Blocksize %lu", smpi_shared_malloc_blocksize);
   /* Create a fd to a new file on disk, make it smpi_shared_malloc_blocksize big, and unlink it.
    * It still exists in memory but not in the file system (thus it cannot be leaked). */
   /* Create bogus file if not done already */
+#ifdef MAP_HUGETLB
   if(use_huge_page && smpi_shared_malloc_bogusfile_huge_page == -1) {
     const char *const array[] = {huge_page_mount_point, "simgrid-shmalloc-XXXXXX", nullptr};
     char *huge_page_filename = xbt_str_join_array(array, "/");
   if(use_huge_page && smpi_shared_malloc_bogusfile_huge_page == -1) {
     const char *const array[] = {huge_page_mount_point, "simgrid-shmalloc-XXXXXX", nullptr};
     char *huge_page_filename = xbt_str_join_array(array, "/");
@@ -263,6 +275,7 @@ void* smpi_shared_malloc_partial(size_t size, size_t* shared_block_offsets, int
     unlink(huge_page_filename);
     xbt_free(huge_page_filename);
   }
     unlink(huge_page_filename);
     xbt_free(huge_page_filename);
   }
+#endif
   if(smpi_shared_malloc_bogusfile == -1) {
     char *name                   = xbt_strdup("/tmp/simgrid-shmalloc-XXXXXX");
     smpi_shared_malloc_bogusfile = mkstemp(name);
   if(smpi_shared_malloc_bogusfile == -1) {
     char *name                   = xbt_strdup("/tmp/simgrid-shmalloc-XXXXXX");
     smpi_shared_malloc_bogusfile = mkstemp(name);
@@ -279,8 +292,10 @@ void* smpi_shared_malloc_partial(size_t size, size_t* shared_block_offsets, int
   int mmap_base_flag = MAP_FIXED | MAP_SHARED | MAP_POPULATE;
   int mmap_flag = mmap_base_flag;
   int huge_fd = use_huge_page ? smpi_shared_malloc_bogusfile_huge_page : smpi_shared_malloc_bogusfile;
   int mmap_base_flag = MAP_FIXED | MAP_SHARED | MAP_POPULATE;
   int mmap_flag = mmap_base_flag;
   int huge_fd = use_huge_page ? smpi_shared_malloc_bogusfile_huge_page : smpi_shared_malloc_bogusfile;
+#ifdef MAP_HUGETLB
   if(use_huge_page)
     mmap_flag |= MAP_HUGETLB;
   if(use_huge_page)
     mmap_flag |= MAP_HUGETLB;
+#endif
 
   XBT_DEBUG("global shared allocation, begin mmap");
 
 
   XBT_DEBUG("global shared allocation, begin mmap");