tensorflow 启用可插拔设备的C/C++客户端

vngu2lb8  于 5个月前  发布在  C/C++
关注(0)|答案(3)|浏览(121)

问题类型

功能请求

来源

source

Tensorflow版本

2.10

自定义代码

OS平台和发行版

  • 无响应*

移动设备

  • 无响应*

Python版本

  • 无响应*

Bazel版本

  • 无响应*

GCC/编译器版本

  • 无响应*

CUDA/cuDNN版本

  • 无响应*

GPU型号和内存

  • 无响应*

当前行为?

There is no TF_LoadPluggableDeviceLibrary defination in libtensorflow_cc.so.
TF_LoadPluggableDeviceLibrary is needed to load PluggableDevice library.

重现问题的独立代码

Add ""//tensorflow/c:c_api_experimental"" deps in "tensorflow_cc" build, tensorflow/BUILD
Refer to https://github.com/feng-intel/intel-itex/tree/main/enable_cc_example  -> 2. Build libtensorflow_cc.so

相关日志输出

  • 无响应*
drnojrws

drnojrws1#

你好,feng-intel!
感谢您提出的功能请求。
@SuryanarayanaY !
您能否查看一下这个功能请求?
谢谢!

m1m5dgzv

m1m5dgzv2#

你好,feng-intel,
抱歉造成延误。我看到了tensorflow/build文件中的以下内容。

deps = [
        "//tensorflow/c:c_api",
        "//tensorflow/c:c_api_experimental",
        "//tensorflow/c:exported_symbols.lds",
        "//tensorflow/c:version_script.lds",
        "//tensorflow/c/eager:c_api",
        "//tensorflow/c/eager:c_api_experimental",
        "//tensorflow/core:distributed_tensorflow_dependencies",
        "//tensorflow/core:tensorflow",
    ]

你是否希望在tensorflow/cc/build文件中也添加相同的依赖项,以便启用可插拔设备的C/C++客户端?请确认。

chy5wohz

chy5wohz3#

TF_LoadPluggableDeviceLibrary在libtensorflow_cc.so中未定义。可插拔设备需要TF_LoadPluggableDeviceLibrary来加载库并初始化。要求是libtensorflow_cc.so具有TF_LoadPluggableDeviceLibrary定义。以下更改对我有效。谢谢。

diff --git a/tensorflow/BUILD b/tensorflow/BUILD
index 19ee8000206..77d8c714729 100644
--- a/tensorflow/BUILD
+++ b/tensorflow/BUILD
@@ -1187,6 +1187,7 @@ tf_cc_shared_library(
     visibility = ["//visibility:public"],
     win_def_file = ":tensorflow_filtered_def_file",
     deps = [
+        "//tensorflow/c:c_api_experimental",
         "//tensorflow/c:c_api",
         "//tensorflow/c:env",
         "//tensorflow/c:kernels",

相关问题