2

我正在 tensorflow 框架中编写 C++ 代码,我想使用使用 makefile 编写的动态库。在源代码中,我放置了头文件的路径:

#include "tensorflow/cc/include/libtrading/proto/tf_fix_client.h"

使用一个名为 fix_client(int argc, char **argv) 的函数并在 BUILD 文件中放置动态库的路径,名为 libtrading.so:

cc_binary(
    name = "session",
    srcs = ["work/session.cc"],
    copts = tf_copts(),
    linkopts = [
        "-lpthread",
        "-lm",
        #for libtrading
        "-L/home/alessandro_mercadante/tensor_flows/tensorflow/tensorflow/cc/include/",
        "-ltrading",
    ],
    ...

bazel-build 检索到以下错误:

bazel-out/local_linux-opt/bin/tensorflow/cc/_objs/session/tensorflow/cc/work/session.o: In function `main':
session.cc:(.text.startup.main+0x2b): undefined reference to `fix_client(int, char**)'
collect2: error: ld returned 1 exit status
4

2 回答 2

2

Bazel 要求声明所有依赖项,因此 TensorFlow 库应该在您的 deps 属性中。看起来您的目标并非如此(尤其是 tensorflow 包含的标志不合适)。

在快速查看 TensorFlow 构建文件后,我会说它需要以下 deps 属性:

 deps = [
     "//tensorflow/cc:cc_ops",
     "//tensorflow/core:kernels",
     "//tensorflow/core:tensorflow",
 ],

但是我对TensorFlow本身真的很陌生。

你的 cc_binary 的 deps 属性是什么?

于 2015-11-25T09:13:43.047 回答
2

我发现了问题。bazel 中库的链接是正确的:问题是 libtrading 是一个 C 库,而 tensorflow 是在 c++ 环境中构建的:所有链接的函数都需要包含在以下守卫中:

#ifdef __cplusplus
extern "C" {
#endif
...
#ifdef __cplusplus
}
#endif
于 2015-11-25T10:41:36.063 回答