memory.h

//===-- include/flang/Runtime/CUDA/memory.h ---------------------*- C++ -*-===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
 
#ifndef FORTRAN_RUNTIME_CUDA_MEMORY_H_
#define FORTRAN_RUNTIME_CUDA_MEMORY_H_
 
#include "flang/Runtime/descriptor-consts.h"
#include "flang/Runtime/entry-names.h"
#include <cstddef>
 
namespace Fortran::runtime::cuda {
 
extern "C" {
 
void *RTDECL(CUFMemAlloc)(std::size_t bytes, unsigned type,
    const char *sourceFile = nullptr, int sourceLine = 0);
 
void RTDECL(CUFMemFree)(void *devicePtr, unsigned type,
    const char *sourceFile = nullptr, int sourceLine = 0);
 
void RTDECL(CUFMemsetDescriptor)(Descriptor *desc, void *value,
    const char *sourceFile = nullptr, int sourceLine = 0);
 
void RTDECL(CUFDataTransferPtrPtr)(void *dst, void *src, std::size_t bytes,
    unsigned mode, const char *sourceFile = nullptr, int sourceLine = 0);
 
void RTDECL(CUFDataTransferPtrDesc)(void *dst, Descriptor *src,
    std::size_t bytes, unsigned mode, const char *sourceFile = nullptr,
    int sourceLine = 0);
 
void RTDECL(CUFDataTransferDescDesc)(Descriptor *dst, Descriptor *src,
    unsigned mode, const char *sourceFile = nullptr, int sourceLine = 0);
 
void RTDECL(CUFDataTransferCstDesc)(Descriptor *dst, Descriptor *src,
    unsigned mode, const char *sourceFile = nullptr, int sourceLine = 0);
 
void RTDECL(CUFDataTransferDescDescNoRealloc)(Descriptor *dst, Descriptor *src,
    unsigned mode, const char *sourceFile = nullptr, int sourceLine = 0);
 
void RTDECL(CUFDataTransferGlobalDescDesc)(Descriptor *dst, Descriptor *src,
    unsigned mode, const char *sourceFile = nullptr, int sourceLine = 0);
 
} // extern "C"
} // namespace Fortran::runtime::cuda
#endif // FORTRAN_RUNTIME_CUDA_MEMORY_H_