Question

我不确定如何调试（可能是使用GDB）SWIG接口背后的Python代码。

我可以使用ipdb来观察Tensorflow的Python代码一直到SWIG包装器的执行（例如tf_session.TF_Run中的session.py），但是我想调试C ++代码在SWIG界面后面。

据推测，我使用bazel build --config debug构建了Tensorflow，但是当从Python界面调用时，如何将gdb附加到生成的代码中？

Answer 1

TensorFlow的C ++代码在与调用它的Python代码相同的进程中执行（或者，如果您使用的是分布式版本，则在与创建{{1}的Python程序相同的进程中执行}）。

Python和C ++之间最简单的接口是tensor_c_api.h中的纯C API。要拦截其中一个调用，可以将tf.GrpcServer附加到运行TensorFlow的Python解释器的进程ID，并在其中一个函数上创建断点。

例如，使用交互式Python会话，在第一个终端输入：

gdb

然后，在另一个终端中，启动$ python >>> import tensorflow >>> import os >>> os.getpid() 14680：

gdb

回到Python解释器，创建一个新会话：

$ gdb -p 14680
[...]
(gdb) break TF_NewSession
Breakpoint 1 at 0x7f15f450a4d0
(gdb) continue
Continuing.

解释器将暂停，您的调试器将打印如下内容：

>>> sess = tf.Session()

您现在可以使用Breakpoint 1, 0x00007f15f450a4d0 in TF_NewSession () from [...]/tensorflow/python/_pywrap_tensorflow.so (gdb) backtrace #0 0x00007f15f450a4d0 in TF_NewSession () from [...]/tensorflow/python/_pywrap_tensorflow.so #1 0x00007f15f3ac5cdb in _wrap_TF_NewSession () from [...]/tensorflow/python/_pywrap_tensorflow.so #2 0x000000000049968d in PyEval_EvalFrameEx () #3 0x00000000004a090c in PyEval_EvalCodeEx () #4 0x0000000000499a52 in PyEval_EvalFrameEx () [...]的全部功能来调试TensorFlow。

调试Tensorflow的SWIG接口背后的C ++代码

1 个答案: