src/gmxlib/cuda_tools/pmalloc_cuda.cu

   1 /*
   2  * This file is part of the GROMACS molecular simulation package.
   3  *
   4  * Copyright (c) 1991-2000, University of Groningen, The Netherlands.
   5  * Copyright (c) 2001-2012, The GROMACS development team,
   6  * check out http://www.gromacs.org for more information.
   7  * Copyright (c) 2012, by the GROMACS development team, led by
   8  * David van der Spoel, Berk Hess, Erik Lindahl, and including many
   9  * others, as listed in the AUTHORS file in the top-level source
  10  * directory and at http://www.gromacs.org.
  11  *
  12  * GROMACS is free software; you can redistribute it and/or
  13  * modify it under the terms of the GNU Lesser General Public License
  14  * as published by the Free Software Foundation; either version 2.1
  15  * of the License, or (at your option) any later version.
  16  *
  17  * GROMACS is distributed in the hope that it will be useful,
  18  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  19  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  20  * Lesser General Public License for more details.
  21  *
  22  * You should have received a copy of the GNU Lesser General Public
  23  * License along with GROMACS; if not, see
  24  * http://www.gnu.org/licenses, or write to the Free Software Foundation,
  25  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA.
  26  *
  27  * If you want to redistribute modifications to GROMACS, please
  28  * consider that scientific software is very special. Version
  29  * control is crucial - bugs must be traceable. We will be happy to
  30  * consider code for inclusion in the official distribution, but
  31  * derived work must not be called official GROMACS. Details are found
  32  * in the README & COPYING files - if they are missing, get the
  33  * official version at http://www.gromacs.org.
  34  *
  35  * To help us fund GROMACS development, we humbly ask that you cite
  36  * the research papers on the package. Check out http://www.gromacs.org.
  37  */
  38
  39 #include <stdlib.h>
  40
  41 #include "gmx_fatal.h"
  42
  43 #include "cudautils.cuh"
  44 #include "pmalloc_cuda.h"
  45
  46 /*! Allocates nbytes of page-locked memory.
  47  *  This memory should always be freed using pfree (or with the page-locked
  48  *  free functions provied by the CUDA library).
  49  */
  50 void pmalloc(void **h_ptr, size_t nbytes)
  51 {
  52     cudaError_t stat;
  53     char        strbuf[STRLEN];
  54     int         flag = cudaHostAllocDefault;
  55
  56     if (nbytes == 0)
  57     {
  58         *h_ptr = NULL;
  59         return;
  60     }
  61
  62     CU_CHECK_PREV_ERR();
  63
  64     stat = cudaMallocHost(h_ptr, nbytes, flag);
  65     sprintf(strbuf, "cudaMallocHost of size %d bytes failed", (int)nbytes);
  66     CU_RET_ERR(stat, strbuf);
  67 }
  68
  69 /*! Allocates nbytes of page-locked memory with write-combining.
  70  *  This memory should always be freed using pfree (or with the page-locked
  71  *  free functions provied by the CUDA library).
  72  */
  73 void pmalloc_wc(void **h_ptr, size_t nbytes)
  74 {
  75     cudaError_t stat;
  76     char        strbuf[STRLEN];
  77     int         flag = cudaHostAllocDefault || cudaHostAllocWriteCombined;
  78
  79     if (nbytes == 0)
  80     {
  81         *h_ptr = NULL;
  82         return;
  83     }
  84
  85     CU_CHECK_PREV_ERR();
  86
  87     stat = cudaMallocHost(h_ptr, nbytes, flag);
  88     sprintf(strbuf, "cudaMallocHost of size %d bytes failed", (int)nbytes);
  89     CU_RET_ERR(stat, strbuf);
  90 }
  91
  92 /*! Frees page locked memory allocated with pmalloc.
  93  *  This function can safely be called also with a pointer to a page-locked
  94  *  memory allocated directly with CUDA API calls.
  95  */
  96 void pfree(void *h_ptr)
  97 {
  98     cudaError_t stat;
  99
 100     if (h_ptr == NULL)
 101     {
 102         return;
 103     }
 104
 105     CU_CHECK_PREV_ERR();
 106
 107     stat = cudaFreeHost(h_ptr);
 108     CU_RET_ERR(stat, "cudaFreeHost failed");
 109 }