cuda_api.h

Go to the documentation of this file.

/*
 *  SPDX-FileCopyrightText: Copyright 2021, Siavash Ameli <sameli@berkeley.edu>
 *  SPDX-License-Identifier: BSD-3-Clause
 *  SPDX-FileType: SOURCE
 *
 *  This program is free software: you can redistribute it and/or modify it
 *  under the terms of the license found in the LICENSE.txt file in the root
 *  directory of this source tree.
 */
 
 
#ifndef _CUDA_UTILITIES_CUDA_API_H_
#define _CUDA_UTILITIES_CUDA_API_H_
 
 
// =======
// Headers
// =======
 
#include <cuda_runtime_api.h>  // cudaError_t, cudaMalloc, cudaMemcpy,
                               // cudaSuccess, cudaFree
 
 
// ==========
// Cuda Tools
// ==========
 
 
template<typename ArrayType>
class CudaAPI
{
    public:
 
        // alloc 1
        static ArrayType* alloc(const size_t array_size);
 
        // alloc 2
        static void alloc(
                ArrayType*& device_array,
                const size_t array_size);
 
        // alloc bytes
        static void alloc_bytes(
                void*& device_array,
                const size_t num_bytes);
 
        // copy to device
        static void copy_to_device(
                const ArrayType* host_array,
                const size_t array_size,
                ArrayType* device_array);
 
        // del
        static void del(void* device_array);
 
        // set device
        static void set_device(int device_id);
 
        // get device
        static int get_device();
};
 
#endif  // _CUDA_UTILITIES_CUDA_API_H_