使用相互依赖性和外部变量将CUDA代码与C代码链接

时间:2019-12-09 08:39:45

标签: c cuda g++ circular-dependency nvcc

我有以下相互依赖的文件:

some_code_main.c

#include "utils.h"
#include "cmdline.h"

FILE log;
struct gengetopt_args_info args_info;

extern int abc;

int main(int argc, char *argv[]) {

    initialize_files(&log, &error_log);  // Some helper function

    cmdline_parser(argc, argv, &args_info); // From cmdline.h

    some_utility_function();    // From utils.h

    cuda_function();    // From cuda_code.cu
}

utils.c

int abc;
extern FILE *LOG;

void some_utility_function() {
    // Calculate abc.

    fprintf(LOG, "abc is %d", abc);
}

cuda_code.cu

extern int abc; // From utils.c
extern struct gengetopt_args_info args_info;
extern FILE *LOG;

__global__ void some_kernel(char * input, int some_value, int dev_abc) {
    // Process input.
}


void cuda_function() {

    char * input;
    char * dev_input;

    cudaMalloc();

    int some_value = args_info.some_value;

    some_kernel<<<5, 5>>>( dev_input, some_value, abc );

    cudaMemcpy( input, dev_input, some_size), cudaMemcpyDeviceToHost );

    cudaFree()

    fprintf(LOG, "%s", input);

}

因此some_code_main.c取决于utils.c和cuda_code.cu; utils.c取决于some_code_main.c; cuda_code.c同时依赖some_code_main.c和utils.c。

如果没有这样的循环依赖关系,我知道可以通过在some_code_main.c中为cuda_function()定义一个原型,然后使用诸如以下的代码来编译代码:

all:
    some_code-mt

some_code-mt: cuda_code.o
    g++ -o some_code utils.c some_code_main.c cuda_code.o

cuda_code.o:
    nvcc -c cuda_code.cu

但是由于cuda_code.cu(与nvcc编译)需要some_code_main.c和utils.c(均由g ++编译),而some_code_main.c需要cuda-code.cu,因此我如何确保满足所有依赖性,并且代码编译正确吗?

1 个答案:

答案 0 :(得分:0)

正如评论中指出的那样,我只需要手动指定CUDA库和路径即可,并且运行良好:

g++ -o some_code utils.c some_code_main.c cuda_code.o -L/usr/local/cuda-10.0/lib64 -lcudart -pthread