Cuda NPP库的静态包装器

Question

我遇到了一个问题，试图为一些NPP icc函数创建静态包装器，以便从cGo (Golang)环境中调用它们。

我能够使用提供的静态NPP库创建和编译(C语言)可执行文件，并且它工作得很好。然后我将它编译为一个库(带有必要的标志)。但是，当我尝试从另一个可执行文件(纯C或cGo)链接该库时，我收到错误消息“未定义的对nppiYCbCr420ToRGB_8u_P3C3R的引用”。我哪里做错了？

wrapper.c

#include <nppi_color_conversion.h>
#include <cuda_runtime_api.h>

int YCbCr420ToRGB() {
  NppiSize oSizeROI;

  const Npp8u * const pSrc[3];
  int cSrcStep[3];
  Npp8u *cDst;

  NppStatus ret = nppiYCbCr420ToRGB_8u_P3C3R(pSrc, cSrcStep, cDst, 0, oSizeROI);

  return (int)ret; // ret = 14
}

build.sh

nvcc nppGo.c -lib -lnppicc_static -lnppc_static -lculibos -lcudart_static -lpthread -ldl -lrt -I /usr/local/cuda-10.0/include -L /usr/local/cuda-10.0/lib64 -o libnppGo

caller.c

#include <stdio.h>
#include "nppGo.h"

int main() {

  int ret = YCbCr420ToRGB();
  printf("Return code is: %d\n", ret);

return (int)ret;
}

buildtest.sh

nvcc caller.c -L. -lnppGo -o nppGo

最后我得到了这个错误信息

./libnppGo.a(tmpxft_0000204d_00000000-2_nppGo.o): In function `YCbCr420ToRGB':
nppGo.c:(.text+0xf9): undefined reference to `nppiYCbCr420ToRGB_8u_P3C3R'
collect2: error: ld returned 1 exit status

我还尝试使用另一个链接器/编译器，结果相同：

g++ -c nppGo.c -I /usr/local/cuda-10.0/include
ar rcs nppGo.a nppGo.o libnppicc_static.a libnppc_static.a libculibos.a libcudart_static.a libdl.a

Answer 1

以下合并静态库的命令可能无法正常工作。

ar rcs nppGo.a nppGo.o libnppicc_static.a libnppc_static.a libculibos.a libcudart_static.a libdl.a

相反，使用MRI脚本来合并这些静态库。我想创建nppGo.a使用以下脚本，让我们将其命名为LibnppGo.mri

在此之前，我们需要为nppGo.o创建一个临时静态库。

ar rcs nppGo_temp.a nppGo.o

LibnppGo.mri看起来会像这样：

create nppGo.a
addlib nppGo_temp.a
addlib libnppicc_static.a
addlib libnppc_static.a
addlib libculibos.a 
addlib libcudart_static.a
addlib libdl.a
save
end

并以如下方式执行ar：

ar -M <LibnppGo.mri

你应该得到一个正确的最终的静态库，你可以正确地链接它。