sp1-gpu-sys 6.2.2

FFI bindings and CUDA build system for SP1-GPU
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33

#pragma once

struct RustCudaError {
    const char* message;
};

typedef struct RustCudaError rustCudaError_t;

extern "C" const rustCudaError_t CUDA_SUCCESS_CSL;

extern "C" const rustCudaError_t CUDA_OUT_OF_MEMORY;

extern "C" const rustCudaError_t CUDA_ERROR_NOT_READY_SLOP;

#define CUDA_UNWRAP(expr)                                                                          \
    do {                                                                                           \
        cudaError_t code = expr;                                                                   \
        if (code != cudaSuccess) {                                                                 \
            std::stringstream ss;                                                                  \
            ss << __FILE__ << "(" << __LINE__ << ")";                                              \
            std::string file_and_line;                                                             \
            ss >> file_and_line;                                                                   \
            throw thrust::system_error(code, thrust::cuda_category(), file_and_line);              \
        }                                                                                          \
    } while (0)

#define CUDA_OK(expr)                                                                              \
    do {                                                                                           \
        cudaError_t code = expr;                                                                   \
        if (code != cudaSuccess) {                                                                 \
            return rustCudaError_t{.message = cudaGetErrorString(code)};                           \
        }                                                                                          \
    } while (0)