54f37133b7
Mostly mechanical changes in preparation of extracting the Flang-RT "subproject" in #110217. This PR intends to only move pre-existing files to the new folder structure, with no behavioral change. Common files (headers, testing, cmake) shared by Flang-RT and Flang remain in `flang/`. Some cosmetic changes and files paths were necessary: * Relative paths to the new path for the source files and `add_subdirectory`. * Add the new location's include directory to `include_directories` * The unittest/Evaluate directory has unitests for flang-rt and Flang. A new `CMakeLists.txt` was introduced for the flang-rt tests. * Change the `#include` paths relative to the include directive * clang-format on the `#include` directives * Since the paths are part if the copyright header and include guards, a script was used to canonicalize those * `test/Runtime` and runtime tests in `test/Driver` are moved, but the lit.cfg.py mechanism to execute the will only be added in #110217.
48 lines
1.6 KiB
C++
48 lines
1.6 KiB
C++
//===-- lib/cuda/registration.cpp -------------------------------*- 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
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "flang/Runtime/CUDA/registration.h"
|
|
#include "flang-rt/runtime/terminator.h"
|
|
#include "flang/Runtime/CUDA/common.h"
|
|
|
|
#include "cuda_runtime.h"
|
|
|
|
namespace Fortran::runtime::cuda {
|
|
|
|
extern "C" {
|
|
|
|
extern void **__cudaRegisterFatBinary(void *);
|
|
extern void __cudaRegisterFatBinaryEnd(void *);
|
|
extern void __cudaRegisterFunction(void **fatCubinHandle, const char *hostFun,
|
|
char *deviceFun, const char *deviceName, int thread_limit, uint3 *tid,
|
|
uint3 *bid, dim3 *bDim, dim3 *gDim, int *wSize);
|
|
extern void __cudaRegisterVar(void **fatCubinHandle, char *hostVar,
|
|
const char *deviceAddress, const char *deviceName, int ext, size_t size,
|
|
int constant, int global);
|
|
|
|
void *RTDECL(CUFRegisterModule)(void *data) {
|
|
void **fatHandle{__cudaRegisterFatBinary(data)};
|
|
__cudaRegisterFatBinaryEnd(fatHandle);
|
|
return fatHandle;
|
|
}
|
|
|
|
void RTDEF(CUFRegisterFunction)(
|
|
void **module, const char *fctSym, char *fctName) {
|
|
__cudaRegisterFunction(module, fctSym, fctName, fctName, -1, (uint3 *)0,
|
|
(uint3 *)0, (dim3 *)0, (dim3 *)0, (int *)0);
|
|
}
|
|
|
|
void RTDEF(CUFRegisterVariable)(
|
|
void **module, char *varSym, const char *varName, int64_t size) {
|
|
__cudaRegisterVar(module, varSym, varName, varName, 0, size, 0, 0);
|
|
}
|
|
|
|
} // extern "C"
|
|
|
|
} // namespace Fortran::runtime::cuda
|