1*9ba83ac0SJeremy L Thompson // Copyright (c) 2017-2026, Lawrence Livermore National Security, LLC and other CEED contributors.
2c9c2c079SJeremy L Thompson // All Rights Reserved. See the top-level LICENSE and NOTICE files for details.
3c9c2c079SJeremy L Thompson //
4c9c2c079SJeremy L Thompson // SPDX-License-Identifier: BSD-2-Clause
5c9c2c079SJeremy L Thompson //
6c9c2c079SJeremy L Thompson // This file is part of CEED: http://github.com/ceed
7c9c2c079SJeremy L Thompson
82b730f8bSJeremy L Thompson #include "ceed-cuda-compile.h"
92b730f8bSJeremy L Thompson
1049aac155SJeremy L Thompson #include <ceed.h>
11c9c2c079SJeremy L Thompson #include <ceed/backend.h>
12c9c2c079SJeremy L Thompson #include <ceed/jit-tools.h>
13c9c2c079SJeremy L Thompson #include <cuda_runtime.h>
142027fb9dSSirAlienTheGreat #include <dirent.h>
15c9c2c079SJeremy L Thompson #include <nvrtc.h>
16c9c2c079SJeremy L Thompson #include <stdarg.h>
17fa619eccSJeremy L Thompson #include <stdio.h>
18c9c2c079SJeremy L Thompson #include <string.h>
199b5f41c8SJeremy L Thompson #include <sys/stat.h>
202027fb9dSSirAlienTheGreat #include <sys/types.h>
212b730f8bSJeremy L Thompson
222027fb9dSSirAlienTheGreat #include <cstdlib>
232027fb9dSSirAlienTheGreat #include <fstream>
242027fb9dSSirAlienTheGreat #include <iostream>
252b730f8bSJeremy L Thompson #include <sstream>
262027fb9dSSirAlienTheGreat #include <string>
272b730f8bSJeremy L Thompson
28c9c2c079SJeremy L Thompson #include "ceed-cuda-common.h"
29c9c2c079SJeremy L Thompson
30c9c2c079SJeremy L Thompson #define CeedChk_Nvrtc(ceed, x) \
31c9c2c079SJeremy L Thompson do { \
32c9c2c079SJeremy L Thompson nvrtcResult result = static_cast<nvrtcResult>(x); \
332b730f8bSJeremy L Thompson if (result != NVRTC_SUCCESS) return CeedError((ceed), CEED_ERROR_BACKEND, nvrtcGetErrorString(result)); \
34c9c2c079SJeremy L Thompson } while (0)
35c9c2c079SJeremy L Thompson
362b730f8bSJeremy L Thompson #define CeedCallNvrtc(ceed, ...) \
372b730f8bSJeremy L Thompson do { \
382b730f8bSJeremy L Thompson int ierr_q_ = __VA_ARGS__; \
392b730f8bSJeremy L Thompson CeedChk_Nvrtc(ceed, ierr_q_); \
406574a04fSJeremy L Thompson } while (0)
412b730f8bSJeremy L Thompson
422027fb9dSSirAlienTheGreat #define CeedCallSystem(ceed, command, message) CeedCallBackend(CeedCallSystem_Core(ceed, command, message))
432027fb9dSSirAlienTheGreat
442027fb9dSSirAlienTheGreat //------------------------------------------------------------------------------
452027fb9dSSirAlienTheGreat // Call system command and capture stdout + stderr
462027fb9dSSirAlienTheGreat //------------------------------------------------------------------------------
CeedCallSystem_Core(Ceed ceed,const char * command,const char * message)472027fb9dSSirAlienTheGreat static int CeedCallSystem_Core(Ceed ceed, const char *command, const char *message) {
48fa619eccSJeremy L Thompson CeedDebug(ceed, "Running command:\n$ %s", command);
492027fb9dSSirAlienTheGreat FILE *output_stream = popen((command + std::string(" 2>&1")).c_str(), "r");
502027fb9dSSirAlienTheGreat
51fa619eccSJeremy L Thompson CeedCheck(output_stream != nullptr, ceed, CEED_ERROR_BACKEND, "Failed to %s\ncommand:\n$ %s", message, command);
522027fb9dSSirAlienTheGreat
53fa619eccSJeremy L Thompson char line[CEED_MAX_RESOURCE_LEN] = "";
54fa619eccSJeremy L Thompson std::string output = "";
552027fb9dSSirAlienTheGreat
56fa619eccSJeremy L Thompson while (fgets(line, sizeof(line), output_stream) != nullptr) {
57fa619eccSJeremy L Thompson output += line;
582027fb9dSSirAlienTheGreat }
59fa619eccSJeremy L Thompson CeedDebug(ceed, "output:\n%s\n", output.c_str());
60fa619eccSJeremy L Thompson CeedCheck(pclose(output_stream) == 0, ceed, CEED_ERROR_BACKEND, "Failed to %s\ncommand:\n$ %s\nerror:\n%s", message, command, output.c_str());
612027fb9dSSirAlienTheGreat return CEED_ERROR_SUCCESS;
622027fb9dSSirAlienTheGreat }
632027fb9dSSirAlienTheGreat
64c9c2c079SJeremy L Thompson //------------------------------------------------------------------------------
65c9c2c079SJeremy L Thompson // Compile CUDA kernel
66c9c2c079SJeremy L Thompson //------------------------------------------------------------------------------
672027fb9dSSirAlienTheGreat using std::ifstream;
682027fb9dSSirAlienTheGreat using std::ofstream;
692027fb9dSSirAlienTheGreat using std::ostringstream;
702027fb9dSSirAlienTheGreat
CeedCompileCore_Cuda(Ceed ceed,const char * source,const bool throw_error,bool * is_compile_good,CUmodule * module,const CeedInt num_defines,va_list args)71ddae5012SJeremy L Thompson static int CeedCompileCore_Cuda(Ceed ceed, const char *source, const bool throw_error, bool *is_compile_good, CUmodule *module,
72ddae5012SJeremy L Thompson const CeedInt num_defines, va_list args) {
73ca735530SJeremy L Thompson size_t ptx_size;
7422070f95SJeremy L Thompson char *ptx;
75a491a57eSJeremy L Thompson const int num_opts = 4;
764753b775SJeremy L Thompson CeedInt num_jit_source_dirs = 0, num_jit_defines = 0;
77b13efd58SJeremy L Thompson const char **opts;
78c9c2c079SJeremy L Thompson nvrtcProgram prog;
79ca735530SJeremy L Thompson struct cudaDeviceProp prop;
80ca735530SJeremy L Thompson Ceed_Cuda *ceed_data;
81ca735530SJeremy L Thompson
82ca735530SJeremy L Thompson cudaFree(0); // Make sure a Context exists for nvrtc
83c9c2c079SJeremy L Thompson
84c9c2c079SJeremy L Thompson std::ostringstream code;
852027fb9dSSirAlienTheGreat bool using_clang;
862027fb9dSSirAlienTheGreat
872027fb9dSSirAlienTheGreat CeedCallBackend(CeedGetIsClang(ceed, &using_clang));
882027fb9dSSirAlienTheGreat
892027fb9dSSirAlienTheGreat CeedDebug256(ceed, CEED_DEBUG_COLOR_SUCCESS,
902027fb9dSSirAlienTheGreat using_clang ? "Compiling CUDA with Clang backend (with Rust QFunction support)"
912027fb9dSSirAlienTheGreat : "Compiling CUDA with NVRTC backend (without Rust QFunction support).\nTo use the Clang backend, set the environment "
922027fb9dSSirAlienTheGreat "variable GPU_CLANG=1");
93c9c2c079SJeremy L Thompson
94c9c2c079SJeremy L Thompson // Get kernel specific options, such as kernel constants
95c9c2c079SJeremy L Thompson if (num_defines > 0) {
96c9c2c079SJeremy L Thompson char *name;
97c9c2c079SJeremy L Thompson int val;
98ca735530SJeremy L Thompson
99c9c2c079SJeremy L Thompson for (int i = 0; i < num_defines; i++) {
100c9c2c079SJeremy L Thompson name = va_arg(args, char *);
101c9c2c079SJeremy L Thompson val = va_arg(args, int);
102c9c2c079SJeremy L Thompson code << "#define " << name << " " << val << "\n";
103c9c2c079SJeremy L Thompson }
104c9c2c079SJeremy L Thompson }
105c9c2c079SJeremy L Thompson
106c9c2c079SJeremy L Thompson // Standard libCEED definitions for CUDA backends
10791adc9c8SJeremy L Thompson code << "#include <ceed/jit-source/cuda/cuda-jit.h>\n\n";
108c9c2c079SJeremy L Thompson
109c9c2c079SJeremy L Thompson // Non-macro options
110b13efd58SJeremy L Thompson CeedCallBackend(CeedCalloc(num_opts, &opts));
111c9c2c079SJeremy L Thompson opts[0] = "-default-device";
1122b730f8bSJeremy L Thompson CeedCallBackend(CeedGetData(ceed, &ceed_data));
1132b730f8bSJeremy L Thompson CeedCallCuda(ceed, cudaGetDeviceProperties(&prop, ceed_data->device_id));
11429ec485eSJed Brown std::string arch_arg =
11529ec485eSJed Brown #if CUDA_VERSION >= 11010
11629ec485eSJed Brown // NVRTC used to support only virtual architectures through the option
11729ec485eSJed Brown // -arch, since it was only emitting PTX. It will now support actual
11829ec485eSJed Brown // architectures as well to emit SASS.
11929ec485eSJed Brown // https://docs.nvidia.com/cuda/cuda-c-best-practices-guide/index.html#dynamic-code-generation
12029ec485eSJed Brown "-arch=sm_"
12129ec485eSJed Brown #else
12229ec485eSJed Brown "-arch=compute_"
12329ec485eSJed Brown #endif
12429ec485eSJed Brown + std::to_string(prop.major) + std::to_string(prop.minor);
125c9c2c079SJeremy L Thompson opts[1] = arch_arg.c_str();
126c9c2c079SJeremy L Thompson opts[2] = "-Dint32_t=int";
127a491a57eSJeremy L Thompson opts[3] = "-DCEED_RUNNING_JIT_PASS=1";
1284753b775SJeremy L Thompson // Additional include dirs
129b13efd58SJeremy L Thompson {
130b13efd58SJeremy L Thompson const char **jit_source_dirs;
131b13efd58SJeremy L Thompson
132b13efd58SJeremy L Thompson CeedCallBackend(CeedGetJitSourceRoots(ceed, &num_jit_source_dirs, &jit_source_dirs));
133b13efd58SJeremy L Thompson CeedCallBackend(CeedRealloc(num_opts + num_jit_source_dirs, &opts));
134b13efd58SJeremy L Thompson for (CeedInt i = 0; i < num_jit_source_dirs; i++) {
1354753b775SJeremy L Thompson std::ostringstream include_dir_arg;
136b13efd58SJeremy L Thompson
1374753b775SJeremy L Thompson include_dir_arg << "-I" << jit_source_dirs[i];
1384753b775SJeremy L Thompson CeedCallBackend(CeedStringAllocCopy(include_dir_arg.str().c_str(), (char **)&opts[num_opts + i]));
139b13efd58SJeremy L Thompson }
140b13efd58SJeremy L Thompson CeedCallBackend(CeedRestoreJitSourceRoots(ceed, &jit_source_dirs));
141b13efd58SJeremy L Thompson }
1424753b775SJeremy L Thompson // User defines
1434753b775SJeremy L Thompson {
1444753b775SJeremy L Thompson const char **jit_defines;
1454753b775SJeremy L Thompson
1464753b775SJeremy L Thompson CeedCallBackend(CeedGetJitDefines(ceed, &num_jit_defines, &jit_defines));
1474753b775SJeremy L Thompson CeedCallBackend(CeedRealloc(num_opts + num_jit_source_dirs + num_jit_defines, &opts));
1484753b775SJeremy L Thompson for (CeedInt i = 0; i < num_jit_defines; i++) {
1494753b775SJeremy L Thompson std::ostringstream define_arg;
1504753b775SJeremy L Thompson
1514753b775SJeremy L Thompson define_arg << "-D" << jit_defines[i];
1524753b775SJeremy L Thompson CeedCallBackend(CeedStringAllocCopy(define_arg.str().c_str(), (char **)&opts[num_opts + num_jit_source_dirs + i]));
1534753b775SJeremy L Thompson }
1544753b775SJeremy L Thompson CeedCallBackend(CeedRestoreJitDefines(ceed, &jit_defines));
1554753b775SJeremy L Thompson }
156c9c2c079SJeremy L Thompson
157c9c2c079SJeremy L Thompson // Add string source argument provided in call
158c9c2c079SJeremy L Thompson code << source;
159c9c2c079SJeremy L Thompson
160c9c2c079SJeremy L Thompson // Compile kernel
161c21e34e2SJeremy L Thompson CeedDebug256(ceed, CEED_DEBUG_COLOR_SUCCESS, "---------- ATTEMPTING TO COMPILE JIT SOURCE ----------\n");
16226ef7cdaSJeremy L Thompson CeedDebug(ceed, "Source:\n%s\n", code.str().c_str());
163c21e34e2SJeremy L Thompson CeedDebug256(ceed, CEED_DEBUG_COLOR_SUCCESS, "---------- END OF JIT SOURCE ----------\n");
1642027fb9dSSirAlienTheGreat
1652027fb9dSSirAlienTheGreat if (!using_clang) {
1662027fb9dSSirAlienTheGreat CeedCallNvrtc(ceed, nvrtcCreateProgram(&prog, code.str().c_str(), NULL, 0, NULL, NULL));
1672027fb9dSSirAlienTheGreat
168bdcc2728SJeremy L Thompson if (CeedDebugFlag(ceed)) {
169bdcc2728SJeremy L Thompson // LCOV_EXCL_START
170c21e34e2SJeremy L Thompson CeedDebug256(ceed, CEED_DEBUG_COLOR_SUCCESS, "---------- JiT COMPILER OPTIONS ----------\n");
171bdcc2728SJeremy L Thompson for (CeedInt i = 0; i < num_opts + num_jit_source_dirs + num_jit_defines; i++) {
172bdcc2728SJeremy L Thompson CeedDebug(ceed, "Option %d: %s", i, opts[i]);
173bdcc2728SJeremy L Thompson }
174bdcc2728SJeremy L Thompson CeedDebug(ceed, "");
175c21e34e2SJeremy L Thompson CeedDebug256(ceed, CEED_DEBUG_COLOR_SUCCESS, "---------- END OF JiT COMPILER OPTIONS ----------\n");
176bdcc2728SJeremy L Thompson // LCOV_EXCL_STOP
177bdcc2728SJeremy L Thompson }
1782027fb9dSSirAlienTheGreat
1794753b775SJeremy L Thompson nvrtcResult result = nvrtcCompileProgram(prog, num_opts + num_jit_source_dirs + num_jit_defines, opts);
180ca735530SJeremy L Thompson
181b13efd58SJeremy L Thompson for (CeedInt i = 0; i < num_jit_source_dirs; i++) {
182b13efd58SJeremy L Thompson CeedCallBackend(CeedFree(&opts[num_opts + i]));
183b13efd58SJeremy L Thompson }
1844753b775SJeremy L Thompson for (CeedInt i = 0; i < num_jit_defines; i++) {
1854753b775SJeremy L Thompson CeedCallBackend(CeedFree(&opts[num_opts + num_jit_source_dirs + i]));
1864753b775SJeremy L Thompson }
187b13efd58SJeremy L Thompson CeedCallBackend(CeedFree(&opts));
188ddae5012SJeremy L Thompson *is_compile_good = result == NVRTC_SUCCESS;
18928c1f747SJeremy L Thompson if (!*is_compile_good) {
190c9c2c079SJeremy L Thompson char *log;
191ca735530SJeremy L Thompson size_t log_size;
192ca735530SJeremy L Thompson
193ca735530SJeremy L Thompson CeedCallNvrtc(ceed, nvrtcGetProgramLogSize(prog, &log_size));
1942b730f8bSJeremy L Thompson CeedCallBackend(CeedMalloc(log_size, &log));
1952b730f8bSJeremy L Thompson CeedCallNvrtc(ceed, nvrtcGetProgramLog(prog, log));
19628c1f747SJeremy L Thompson if (throw_error) {
1972b730f8bSJeremy L Thompson return CeedError(ceed, CEED_ERROR_BACKEND, "%s\n%s", nvrtcGetErrorString(result), log);
19828c1f747SJeremy L Thompson } else {
199c49dc7a7SJeremy L Thompson // LCOV_EXCL_START
20028c1f747SJeremy L Thompson CeedDebug256(ceed, CEED_DEBUG_COLOR_ERROR, "---------- COMPILE ERROR DETECTED ----------\n");
20128c1f747SJeremy L Thompson CeedDebug(ceed, "Error: %s\nCompile log:\n%s\n", nvrtcGetErrorString(result), log);
2022027fb9dSSirAlienTheGreat CeedDebug256(ceed, CEED_DEBUG_COLOR_ERROR, "---------- BACKEND MAY FALLBACK ----------\n");
20328c1f747SJeremy L Thompson CeedCallBackend(CeedFree(&log));
20428c1f747SJeremy L Thompson CeedCallNvrtc(ceed, nvrtcDestroyProgram(&prog));
20528c1f747SJeremy L Thompson return CEED_ERROR_SUCCESS;
206c49dc7a7SJeremy L Thompson // LCOV_EXCL_STOP
20728c1f747SJeremy L Thompson }
208c9c2c079SJeremy L Thompson }
209c9c2c079SJeremy L Thompson
21029ec485eSJed Brown #if CUDA_VERSION >= 11010
21129ec485eSJed Brown CeedCallNvrtc(ceed, nvrtcGetCUBINSize(prog, &ptx_size));
21229ec485eSJed Brown CeedCallBackend(CeedMalloc(ptx_size, &ptx));
21329ec485eSJed Brown CeedCallNvrtc(ceed, nvrtcGetCUBIN(prog, ptx));
21429ec485eSJed Brown #else
2152b730f8bSJeremy L Thompson CeedCallNvrtc(ceed, nvrtcGetPTXSize(prog, &ptx_size));
2162b730f8bSJeremy L Thompson CeedCallBackend(CeedMalloc(ptx_size, &ptx));
2172b730f8bSJeremy L Thompson CeedCallNvrtc(ceed, nvrtcGetPTX(prog, ptx));
21829ec485eSJed Brown #endif
2192b730f8bSJeremy L Thompson CeedCallNvrtc(ceed, nvrtcDestroyProgram(&prog));
220c9c2c079SJeremy L Thompson
2212b730f8bSJeremy L Thompson CeedCallCuda(ceed, cuModuleLoadData(module, ptx));
2222b730f8bSJeremy L Thompson CeedCallBackend(CeedFree(&ptx));
223c9c2c079SJeremy L Thompson return CEED_ERROR_SUCCESS;
2242027fb9dSSirAlienTheGreat } else {
2259b5f41c8SJeremy L Thompson srand(time(NULL));
2269b5f41c8SJeremy L Thompson const int build_id = rand();
2272027fb9dSSirAlienTheGreat
2289b5f41c8SJeremy L Thompson // Create temp dir if needed
2299b5f41c8SJeremy L Thompson {
2309b5f41c8SJeremy L Thompson DIR *dir = opendir("temp");
2319b5f41c8SJeremy L Thompson
2329b5f41c8SJeremy L Thompson if (dir) {
2339b5f41c8SJeremy L Thompson closedir(dir);
2349b5f41c8SJeremy L Thompson } else {
23533cc410dSJeremy L Thompson // In parallel multiple processes may attempt
23633cc410dSJeremy L Thompson // Only one process needs to succeed
2379b5f41c8SJeremy L Thompson mkdir("temp", 0777);
2389b5f41c8SJeremy L Thompson chmod("temp", 0777);
2399b5f41c8SJeremy L Thompson }
2409b5f41c8SJeremy L Thompson }
2419b5f41c8SJeremy L Thompson // Write code to temp file
2429b5f41c8SJeremy L Thompson {
2439b5f41c8SJeremy L Thompson std::string filename = std::string("temp/kernel_") + std::to_string(build_id) + std::string("_0_source.cu");
2449b5f41c8SJeremy L Thompson FILE *file = fopen(filename.c_str(), "w");
2459b5f41c8SJeremy L Thompson
2469b5f41c8SJeremy L Thompson CeedCheck(file, ceed, CEED_ERROR_BACKEND, "Failed to create file. Write access is required for cuda-clang");
2472027fb9dSSirAlienTheGreat fputs(code.str().c_str(), file);
2482027fb9dSSirAlienTheGreat fclose(file);
2499b5f41c8SJeremy L Thompson }
2502027fb9dSSirAlienTheGreat
2512027fb9dSSirAlienTheGreat // Get rust crate directories
2522027fb9dSSirAlienTheGreat const char **rust_source_dirs = nullptr;
2532027fb9dSSirAlienTheGreat int num_rust_source_dirs = 0;
2542027fb9dSSirAlienTheGreat
2552027fb9dSSirAlienTheGreat CeedCallBackend(CeedGetRustSourceRoots(ceed, &num_rust_source_dirs, &rust_source_dirs));
2562027fb9dSSirAlienTheGreat
2572027fb9dSSirAlienTheGreat std::string rust_dirs[10];
2582027fb9dSSirAlienTheGreat
2592027fb9dSSirAlienTheGreat if (num_rust_source_dirs > 0) {
2602027fb9dSSirAlienTheGreat CeedDebug(ceed, "There are %d source dirs, including %s\n", num_rust_source_dirs, rust_source_dirs[0]);
2612027fb9dSSirAlienTheGreat }
2622027fb9dSSirAlienTheGreat
2632027fb9dSSirAlienTheGreat for (CeedInt i = 0; i < num_rust_source_dirs; i++) {
2642027fb9dSSirAlienTheGreat rust_dirs[i] = std::string(rust_source_dirs[i]);
2652027fb9dSSirAlienTheGreat }
2662027fb9dSSirAlienTheGreat
2672027fb9dSSirAlienTheGreat CeedCallBackend(CeedRestoreRustSourceRoots(ceed, &rust_source_dirs));
2682027fb9dSSirAlienTheGreat
2692027fb9dSSirAlienTheGreat char *rust_toolchain = std::getenv("RUST_TOOLCHAIN");
2702027fb9dSSirAlienTheGreat
2712027fb9dSSirAlienTheGreat if (rust_toolchain == nullptr) {
2722027fb9dSSirAlienTheGreat rust_toolchain = (char *)"nightly";
2732027fb9dSSirAlienTheGreat setenv("RUST_TOOLCHAIN", "nightly", 0);
2742027fb9dSSirAlienTheGreat }
2752027fb9dSSirAlienTheGreat
2762027fb9dSSirAlienTheGreat // Compile Rust crate(s) needed
2772027fb9dSSirAlienTheGreat std::string command;
2782027fb9dSSirAlienTheGreat
2792027fb9dSSirAlienTheGreat for (CeedInt i = 0; i < num_rust_source_dirs; i++) {
2802027fb9dSSirAlienTheGreat command = "cargo +" + std::string(rust_toolchain) + " build --release --target nvptx64-nvidia-cuda --config " + rust_dirs[i] +
2812027fb9dSSirAlienTheGreat "/.cargo/config.toml --manifest-path " + rust_dirs[i] + "/Cargo.toml";
2822027fb9dSSirAlienTheGreat CeedCallSystem(ceed, command.c_str(), "build Rust crate");
2832027fb9dSSirAlienTheGreat }
2842027fb9dSSirAlienTheGreat
285fa619eccSJeremy L Thompson // Get Clang version
286fa619eccSJeremy L Thompson bool use_llvm_version = ceed_data->use_llvm_version;
287fa619eccSJeremy L Thompson int llvm_version = ceed_data->llvm_version;
288fa619eccSJeremy L Thompson
289fa619eccSJeremy L Thompson if (llvm_version == 0) {
290fa619eccSJeremy L Thompson command = "$(find $(rustup run " + std::string(rust_toolchain) + " rustc --print sysroot) -name llvm-link) --version";
291fa619eccSJeremy L Thompson CeedDebug(ceed, "Attempting to detect Rust LLVM version.\ncommand:\n$ %s", command.c_str());
292fa619eccSJeremy L Thompson FILE *output_stream = popen((command + std::string(" 2>&1")).c_str(), "r");
293fa619eccSJeremy L Thompson
294fa619eccSJeremy L Thompson CeedCheck(output_stream != nullptr, ceed, CEED_ERROR_BACKEND, "Failed to detect Rust LLVM version");
295fa619eccSJeremy L Thompson
296fa619eccSJeremy L Thompson char line[CEED_MAX_RESOURCE_LEN] = "";
297fa619eccSJeremy L Thompson std::string output = "";
298fa619eccSJeremy L Thompson
299fa619eccSJeremy L Thompson while (fgets(line, sizeof(line), output_stream) != nullptr) {
300fa619eccSJeremy L Thompson output += line;
301fa619eccSJeremy L Thompson }
302fa619eccSJeremy L Thompson CeedDebug(ceed, "output:\n%s", output.c_str());
303fa619eccSJeremy L Thompson CeedCheck(pclose(output_stream) == 0, ceed, CEED_ERROR_BACKEND, "Failed to detect Rust LLVM version\ncommand:\n$ %s\nerror:\n%s",
304fa619eccSJeremy L Thompson command.c_str(), output.c_str());
305fa619eccSJeremy L Thompson
306fa619eccSJeremy L Thompson const char *version_substring = strstr(output.c_str(), "LLVM version ");
307fa619eccSJeremy L Thompson
308fa619eccSJeremy L Thompson version_substring += 13;
309fa619eccSJeremy L Thompson
310fa619eccSJeremy L Thompson char *next_dot = strchr((char *)version_substring, '.');
311fa619eccSJeremy L Thompson
312f03c7eeeSJeremy L Thompson if (next_dot) {
313fa619eccSJeremy L Thompson next_dot[0] = '\0';
314fa619eccSJeremy L Thompson ceed_data->llvm_version = llvm_version = std::stoi(version_substring);
315fa619eccSJeremy L Thompson CeedDebug(ceed, "Rust LLVM version number: %d\n", llvm_version);
316fa619eccSJeremy L Thompson
317fa619eccSJeremy L Thompson command = std::string("clang++-") + std::to_string(llvm_version);
318fa619eccSJeremy L Thompson output_stream = popen((command + std::string(" 2>&1")).c_str(), "r");
319fa619eccSJeremy L Thompson ceed_data->use_llvm_version = use_llvm_version = pclose(output_stream) == 0;
320f03c7eeeSJeremy L Thompson } else {
321f03c7eeeSJeremy L Thompson ceed_data->llvm_version = -1;
322f03c7eeeSJeremy L Thompson ceed_data->use_llvm_version = use_llvm_version = false;
323f03c7eeeSJeremy L Thompson }
324fa619eccSJeremy L Thompson }
325fa619eccSJeremy L Thompson
3262027fb9dSSirAlienTheGreat // Compile wrapper kernel
327fa619eccSJeremy L Thompson command = "clang++" + (use_llvm_version ? (std::string("-") + std::to_string(llvm_version)) : "") + " -flto=thin --cuda-gpu-arch=sm_" +
328fa619eccSJeremy L Thompson std::to_string(prop.major) + std::to_string(prop.minor) + " --cuda-device-only -emit-llvm -S temp/kernel_" + std::to_string(build_id) +
329fa619eccSJeremy L Thompson "_0_source.cu -o temp/kernel_" + std::to_string(build_id) + "_1_wrapped.ll ";
3302027fb9dSSirAlienTheGreat command += opts[4];
3312027fb9dSSirAlienTheGreat CeedCallSystem(ceed, command.c_str(), "JiT kernel source");
33233cc410dSJeremy L Thompson CeedCallSystem(ceed, ("chmod 0777 temp/kernel_" + std::to_string(build_id) + "_1_wrapped.ll").c_str(), "update JiT file permissions");
3332027fb9dSSirAlienTheGreat
334f03c7eeeSJeremy L Thompson // Find Rust's llvm-link tool and run it
3359b5f41c8SJeremy L Thompson command = "$(find $(rustup run " + std::string(rust_toolchain) + " rustc --print sysroot) -name llvm-link) temp/kernel_" +
3369b5f41c8SJeremy L Thompson std::to_string(build_id) +
3379b5f41c8SJeremy L Thompson "_1_wrapped.ll --ignore-non-bitcode --internalize --only-needed -S -o "
3389b5f41c8SJeremy L Thompson "temp/kernel_" +
3399b5f41c8SJeremy L Thompson std::to_string(build_id) + "_2_linked.ll ";
3402027fb9dSSirAlienTheGreat
341f03c7eeeSJeremy L Thompson // Searches for .a files in Rust directory
342f03c7eeeSJeremy L Thompson // Note: Rust crate names may not match the folder they are in
343f03c7eeeSJeremy L Thompson // TODO: If libCEED switches to c++17, use std::filesystem here
3442027fb9dSSirAlienTheGreat for (CeedInt i = 0; i < num_rust_source_dirs; i++) {
3452027fb9dSSirAlienTheGreat std::string dir = rust_dirs[i] + "/target/nvptx64-nvidia-cuda/release";
3462027fb9dSSirAlienTheGreat DIR *dp = opendir(dir.c_str());
3472027fb9dSSirAlienTheGreat
3482027fb9dSSirAlienTheGreat CeedCheck(dp != nullptr, ceed, CEED_ERROR_BACKEND, "Could not open directory: %s", dir.c_str());
3492027fb9dSSirAlienTheGreat struct dirent *entry;
3502027fb9dSSirAlienTheGreat
351fa619eccSJeremy L Thompson // Find files ending in .a
3522027fb9dSSirAlienTheGreat while ((entry = readdir(dp)) != nullptr) {
3532027fb9dSSirAlienTheGreat std::string filename(entry->d_name);
3542027fb9dSSirAlienTheGreat
3552027fb9dSSirAlienTheGreat if (filename.size() >= 2 && filename.substr(filename.size() - 2) == ".a") {
3562027fb9dSSirAlienTheGreat command += dir + "/" + filename + " ";
3572027fb9dSSirAlienTheGreat }
3582027fb9dSSirAlienTheGreat }
3592027fb9dSSirAlienTheGreat closedir(dp);
3602027fb9dSSirAlienTheGreat }
3612027fb9dSSirAlienTheGreat
3622027fb9dSSirAlienTheGreat // Link, optimize, and compile final CUDA kernel
3632027fb9dSSirAlienTheGreat CeedCallSystem(ceed, command.c_str(), "link C and Rust source");
3641a8516d0SJames Wright CeedCallSystem(ceed,
3651a8516d0SJames Wright ("$(find $(rustup run " + std::string(rust_toolchain) +
3661a8516d0SJames Wright " rustc --print sysroot) -name opt) --passes internalize,inline temp/kernel_" + std::to_string(build_id) +
3671a8516d0SJames Wright "_2_linked.ll -o temp/kernel_" + std::to_string(build_id) + "_3_opt.bc")
3682027fb9dSSirAlienTheGreat .c_str(),
3692027fb9dSSirAlienTheGreat "optimize linked C and Rust source");
37033cc410dSJeremy L Thompson CeedCallSystem(ceed, ("chmod 0777 temp/kernel_" + std::to_string(build_id) + "_2_linked.ll").c_str(), "update JiT file permissions");
3712027fb9dSSirAlienTheGreat CeedCallSystem(ceed,
3722027fb9dSSirAlienTheGreat ("$(find $(rustup run " + std::string(rust_toolchain) + " rustc --print sysroot) -name llc) -O3 -mcpu=sm_" +
3739b5f41c8SJeremy L Thompson std::to_string(prop.major) + std::to_string(prop.minor) + " temp/kernel_" + std::to_string(build_id) +
3749b5f41c8SJeremy L Thompson "_3_opt.bc -o temp/kernel_" + std::to_string(build_id) + "_4_final.ptx")
3752027fb9dSSirAlienTheGreat .c_str(),
3762027fb9dSSirAlienTheGreat "compile final CUDA kernel");
37733cc410dSJeremy L Thompson CeedCallSystem(ceed, ("chmod 0777 temp/kernel_" + std::to_string(build_id) + "_4_final.ptx").c_str(), "update JiT file permissions");
3782027fb9dSSirAlienTheGreat
379f03c7eeeSJeremy L Thompson // Load module from final PTX
3809b5f41c8SJeremy L Thompson ifstream ptxfile("temp/kernel_" + std::to_string(build_id) + "_4_final.ptx");
3812027fb9dSSirAlienTheGreat ostringstream sstr;
3822027fb9dSSirAlienTheGreat
3832027fb9dSSirAlienTheGreat sstr << ptxfile.rdbuf();
3842027fb9dSSirAlienTheGreat
3852027fb9dSSirAlienTheGreat auto ptx_data = sstr.str();
3862027fb9dSSirAlienTheGreat ptx_size = ptx_data.length();
3872027fb9dSSirAlienTheGreat
3882027fb9dSSirAlienTheGreat int result = cuModuleLoadData(module, ptx_data.c_str());
3892027fb9dSSirAlienTheGreat
3902027fb9dSSirAlienTheGreat *is_compile_good = result == 0;
3912027fb9dSSirAlienTheGreat if (!*is_compile_good) {
3922027fb9dSSirAlienTheGreat if (throw_error) {
3932027fb9dSSirAlienTheGreat return CeedError(ceed, CEED_ERROR_BACKEND, "Failed to load module data");
3942027fb9dSSirAlienTheGreat } else {
3952027fb9dSSirAlienTheGreat // LCOV_EXCL_START
3962027fb9dSSirAlienTheGreat CeedDebug256(ceed, CEED_DEBUG_COLOR_ERROR, "---------- COMPILE ERROR DETECTED ----------\n");
3972027fb9dSSirAlienTheGreat CeedDebug(ceed, "Error: Failed to load module data");
3982027fb9dSSirAlienTheGreat CeedDebug256(ceed, CEED_DEBUG_COLOR_ERROR, "---------- BACKEND MAY FALLBACK ----------\n");
3992027fb9dSSirAlienTheGreat return CEED_ERROR_SUCCESS;
4002027fb9dSSirAlienTheGreat // LCOV_EXCL_STOP
4012027fb9dSSirAlienTheGreat }
4022027fb9dSSirAlienTheGreat }
4032027fb9dSSirAlienTheGreat }
4042027fb9dSSirAlienTheGreat return CEED_ERROR_SUCCESS;
405c9c2c079SJeremy L Thompson }
406c9c2c079SJeremy L Thompson
CeedCompile_Cuda(Ceed ceed,const char * source,CUmodule * module,const CeedInt num_defines,...)407ddae5012SJeremy L Thompson int CeedCompile_Cuda(Ceed ceed, const char *source, CUmodule *module, const CeedInt num_defines, ...) {
408ddae5012SJeremy L Thompson bool is_compile_good = true;
409ddae5012SJeremy L Thompson va_list args;
410ddae5012SJeremy L Thompson
411ddae5012SJeremy L Thompson va_start(args, num_defines);
41218c38aeeSJeremy L Thompson const CeedInt ierr = CeedCompileCore_Cuda(ceed, source, true, &is_compile_good, module, num_defines, args);
41318c38aeeSJeremy L Thompson
414ddae5012SJeremy L Thompson va_end(args);
41518c38aeeSJeremy L Thompson CeedCallBackend(ierr);
416ddae5012SJeremy L Thompson return CEED_ERROR_SUCCESS;
417ddae5012SJeremy L Thompson }
418ddae5012SJeremy L Thompson
CeedTryCompile_Cuda(Ceed ceed,const char * source,bool * is_compile_good,CUmodule * module,const CeedInt num_defines,...)419ddae5012SJeremy L Thompson int CeedTryCompile_Cuda(Ceed ceed, const char *source, bool *is_compile_good, CUmodule *module, const CeedInt num_defines, ...) {
420ddae5012SJeremy L Thompson va_list args;
421ddae5012SJeremy L Thompson
422ddae5012SJeremy L Thompson va_start(args, num_defines);
42318c38aeeSJeremy L Thompson const CeedInt ierr = CeedCompileCore_Cuda(ceed, source, false, is_compile_good, module, num_defines, args);
42418c38aeeSJeremy L Thompson
425ddae5012SJeremy L Thompson va_end(args);
42618c38aeeSJeremy L Thompson CeedCallBackend(ierr);
427ddae5012SJeremy L Thompson return CEED_ERROR_SUCCESS;
428ddae5012SJeremy L Thompson }
429ddae5012SJeremy L Thompson
430c9c2c079SJeremy L Thompson //------------------------------------------------------------------------------
431c9c2c079SJeremy L Thompson // Get CUDA kernel
432c9c2c079SJeremy L Thompson //------------------------------------------------------------------------------
CeedGetKernel_Cuda(Ceed ceed,CUmodule module,const char * name,CUfunction * kernel)433eb7e6cafSJeremy L Thompson int CeedGetKernel_Cuda(Ceed ceed, CUmodule module, const char *name, CUfunction *kernel) {
4342b730f8bSJeremy L Thompson CeedCallCuda(ceed, cuModuleGetFunction(kernel, module, name));
435c9c2c079SJeremy L Thompson return CEED_ERROR_SUCCESS;
436c9c2c079SJeremy L Thompson }
437c9c2c079SJeremy L Thompson
438b2165e7aSSebastian Grimberg //------------------------------------------------------------------------------
439b2165e7aSSebastian Grimberg // Run CUDA kernel with block size selected automatically based on the kernel
440b2165e7aSSebastian Grimberg // (which may use enough registers to require a smaller block size than the
441b2165e7aSSebastian Grimberg // hardware is capable)
442b2165e7aSSebastian Grimberg //------------------------------------------------------------------------------
CeedRunKernelAutoblockCuda(Ceed ceed,CUfunction kernel,size_t points,void ** args)4432b730f8bSJeremy L Thompson int CeedRunKernelAutoblockCuda(Ceed ceed, CUfunction kernel, size_t points, void **args) {
444c9c2c079SJeremy L Thompson int min_grid_size, max_block_size;
445ca735530SJeremy L Thompson
4462b730f8bSJeremy L Thompson CeedCallCuda(ceed, cuOccupancyMaxPotentialBlockSize(&min_grid_size, &max_block_size, kernel, NULL, 0, 0x10000));
447eb7e6cafSJeremy L Thompson CeedCallBackend(CeedRunKernel_Cuda(ceed, kernel, CeedDivUpInt(points, max_block_size), max_block_size, args));
448ca735530SJeremy L Thompson return CEED_ERROR_SUCCESS;
449c9c2c079SJeremy L Thompson }
450c9c2c079SJeremy L Thompson
451c9c2c079SJeremy L Thompson //------------------------------------------------------------------------------
452c9c2c079SJeremy L Thompson // Run CUDA kernel
453c9c2c079SJeremy L Thompson //------------------------------------------------------------------------------
CeedRunKernel_Cuda(Ceed ceed,CUfunction kernel,const int grid_size,const int block_size,void ** args)454eb7e6cafSJeremy L Thompson int CeedRunKernel_Cuda(Ceed ceed, CUfunction kernel, const int grid_size, const int block_size, void **args) {
455e9c76bddSJeremy L Thompson CeedCallBackend(CeedRunKernelDimShared_Cuda(ceed, kernel, NULL, grid_size, block_size, 1, 1, 0, args));
456c9c2c079SJeremy L Thompson return CEED_ERROR_SUCCESS;
457c9c2c079SJeremy L Thompson }
458c9c2c079SJeremy L Thompson
459c9c2c079SJeremy L Thompson //------------------------------------------------------------------------------
460c9c2c079SJeremy L Thompson // Run CUDA kernel for spatial dimension
461c9c2c079SJeremy L Thompson //------------------------------------------------------------------------------
CeedRunKernelDim_Cuda(Ceed ceed,CUfunction kernel,const int grid_size,const int block_size_x,const int block_size_y,const int block_size_z,void ** args)462eb7e6cafSJeremy L Thompson int CeedRunKernelDim_Cuda(Ceed ceed, CUfunction kernel, const int grid_size, const int block_size_x, const int block_size_y, const int block_size_z,
4632b730f8bSJeremy L Thompson void **args) {
464e9c76bddSJeremy L Thompson CeedCallBackend(CeedRunKernelDimShared_Cuda(ceed, kernel, NULL, grid_size, block_size_x, block_size_y, block_size_z, 0, args));
465c9c2c079SJeremy L Thompson return CEED_ERROR_SUCCESS;
466c9c2c079SJeremy L Thompson }
467c9c2c079SJeremy L Thompson
468c9c2c079SJeremy L Thompson //------------------------------------------------------------------------------
469ea61e9acSJeremy L Thompson // Run CUDA kernel for spatial dimension with shared memory
470c9c2c079SJeremy L Thompson //------------------------------------------------------------------------------
CeedRunKernelDimSharedCore_Cuda(Ceed ceed,CUfunction kernel,CUstream stream,const int grid_size,const int block_size_x,const int block_size_y,const int block_size_z,const int shared_mem_size,const bool throw_error,bool * is_good_run,void ** args)471e9c76bddSJeremy L Thompson static int CeedRunKernelDimSharedCore_Cuda(Ceed ceed, CUfunction kernel, CUstream stream, const int grid_size, const int block_size_x,
472e9c76bddSJeremy L Thompson const int block_size_y, const int block_size_z, const int shared_mem_size, const bool throw_error,
473e9c76bddSJeremy L Thompson bool *is_good_run, void **args) {
474023b8a51Sabdelfattah83 #if CUDA_VERSION >= 9000
475023b8a51Sabdelfattah83 cuFuncSetAttribute(kernel, CU_FUNC_ATTRIBUTE_MAX_DYNAMIC_SHARED_SIZE_BYTES, shared_mem_size);
476023b8a51Sabdelfattah83 #endif
477e9c76bddSJeremy L Thompson CUresult result = cuLaunchKernel(kernel, grid_size, 1, 1, block_size_x, block_size_y, block_size_z, shared_mem_size, stream, args, NULL);
478ca735530SJeremy L Thompson
479c9c2c079SJeremy L Thompson if (result == CUDA_ERROR_LAUNCH_OUT_OF_RESOURCES) {
480c9c2c079SJeremy L Thompson int max_threads_per_block, shared_size_bytes, num_regs;
481ca735530SJeremy L Thompson
4822b730f8bSJeremy L Thompson cuFuncGetAttribute(&max_threads_per_block, CU_FUNC_ATTRIBUTE_MAX_THREADS_PER_BLOCK, kernel);
4832b730f8bSJeremy L Thompson cuFuncGetAttribute(&shared_size_bytes, CU_FUNC_ATTRIBUTE_SHARED_SIZE_BYTES, kernel);
484c9c2c079SJeremy L Thompson cuFuncGetAttribute(&num_regs, CU_FUNC_ATTRIBUTE_NUM_REGS, kernel);
485c49dc7a7SJeremy L Thompson if (throw_error) {
486c9c2c079SJeremy L Thompson return CeedError(ceed, CEED_ERROR_BACKEND,
487c9c2c079SJeremy L Thompson "CUDA_ERROR_LAUNCH_OUT_OF_RESOURCES: max_threads_per_block %d on block size (%d,%d,%d), shared_size %d, num_regs %d",
4882b730f8bSJeremy L Thompson max_threads_per_block, block_size_x, block_size_y, block_size_z, shared_size_bytes, num_regs);
489c49dc7a7SJeremy L Thompson } else {
490c49dc7a7SJeremy L Thompson // LCOV_EXCL_START
491c49dc7a7SJeremy L Thompson CeedDebug256(ceed, CEED_DEBUG_COLOR_ERROR, "---------- LAUNCH ERROR DETECTED ----------\n");
492c49dc7a7SJeremy L Thompson CeedDebug(ceed, "CUDA_ERROR_LAUNCH_OUT_OF_RESOURCES: max_threads_per_block %d on block size (%d,%d,%d), shared_size %d, num_regs %d\n",
493c49dc7a7SJeremy L Thompson max_threads_per_block, block_size_x, block_size_y, block_size_z, shared_size_bytes, num_regs);
494c21e34e2SJeremy L Thompson CeedDebug256(ceed, CEED_DEBUG_COLOR_WARNING, "---------- BACKEND MAY FALLBACK ----------\n");
495c49dc7a7SJeremy L Thompson // LCOV_EXCL_STOP
496ddae5012SJeremy L Thompson }
497c49dc7a7SJeremy L Thompson *is_good_run = false;
498c9c2c079SJeremy L Thompson } else CeedChk_Cu(ceed, result);
499c9c2c079SJeremy L Thompson return CEED_ERROR_SUCCESS;
500c9c2c079SJeremy L Thompson }
5012a86cc9dSSebastian Grimberg
CeedRunKernelDimShared_Cuda(Ceed ceed,CUfunction kernel,CUstream stream,const int grid_size,const int block_size_x,const int block_size_y,const int block_size_z,const int shared_mem_size,void ** args)502e9c76bddSJeremy L Thompson int CeedRunKernelDimShared_Cuda(Ceed ceed, CUfunction kernel, CUstream stream, const int grid_size, const int block_size_x, const int block_size_y,
503ddae5012SJeremy L Thompson const int block_size_z, const int shared_mem_size, void **args) {
504ddae5012SJeremy L Thompson bool is_good_run = true;
505ddae5012SJeremy L Thompson
506e9c76bddSJeremy L Thompson CeedCallBackend(CeedRunKernelDimSharedCore_Cuda(ceed, kernel, stream, grid_size, block_size_x, block_size_y, block_size_z, shared_mem_size, true,
507e9c76bddSJeremy L Thompson &is_good_run, args));
508ddae5012SJeremy L Thompson return CEED_ERROR_SUCCESS;
509ddae5012SJeremy L Thompson }
510ddae5012SJeremy L Thompson
CeedTryRunKernelDimShared_Cuda(Ceed ceed,CUfunction kernel,CUstream stream,const int grid_size,const int block_size_x,const int block_size_y,const int block_size_z,const int shared_mem_size,bool * is_good_run,void ** args)511e9c76bddSJeremy L Thompson int CeedTryRunKernelDimShared_Cuda(Ceed ceed, CUfunction kernel, CUstream stream, const int grid_size, const int block_size_x, const int block_size_y,
512ddae5012SJeremy L Thompson const int block_size_z, const int shared_mem_size, bool *is_good_run, void **args) {
513e9c76bddSJeremy L Thompson CeedCallBackend(CeedRunKernelDimSharedCore_Cuda(ceed, kernel, stream, grid_size, block_size_x, block_size_y, block_size_z, shared_mem_size, false,
514e9c76bddSJeremy L Thompson is_good_run, args));
515ddae5012SJeremy L Thompson return CEED_ERROR_SUCCESS;
516ddae5012SJeremy L Thompson }
517ddae5012SJeremy L Thompson
5182a86cc9dSSebastian Grimberg //------------------------------------------------------------------------------
519