如何通过在一种方法中两次调用一个函数来编译cuda代码?

How to compile cuda code with calling one function twice inside one method?

我正在尝试编译这段代码:

struct foo {
    unsigned long long x0;
};

//__device__ __noinline__ foo bar(foo a, foo b){  // << try this
__device__ foo bar(foo a, foo b){
    foo r;
    asm(
    ".reg .u64 my_cool_var;\n\t"
    "add.cc.u64 %0, %1, %2;\n\t"
    : "=l"(r.x0)
    : "l"(a.x0)
      "l"(b.x0)
    );
    return r;
}

__device__ foo func_with2call(foo x, foo y){
    foo res = bar(x, y);
    foo iy =  bar(x, res);
    return iy;
}

__global__ void cuda_test1(foo *a, foo *b, foo *r) {
    *r = func_with2call(*a, *b);
}

编译器输出:

ptxas /tmp/tmpxft_000010f5_00000000-6_main.ptx, line 38; error   : Duplicate definition of variable 'my_cool_var'
ptxas fatal   : Ptx assembly aborted due to errors

我知道,这是代码内联的问题。例如,如果我用 __noinline__ 属性编译 bar 函数,那么就没有错误。有没有办法保持内联(除了用不同的内部变量名复制 bar 函数代码),但仍然调用 bar 函数两次?

此限制在 Inline PTX Assembly Guide 中讨论。您可以通过将每个定义强制到其自己的范围内来解决它,例如:

__device__ foo bar(foo a, foo b){
    foo r;
    asm(
    "{.reg .u64 my_cool_var;\n\t"
    "add.cc.u64 %0, %1, %2;\n\t"
    "}"
    : "=l"(r.x0)
    : "l"(a.x0)
      "l"(b.x0)
    );
    return r;
}

这将安全地内联而不会发生冲突。