python - 不能将 line_profiler 与 Cython 一起使用

Question

基于对 this question 的回答我正在尝试使用 line_profiler具有 cythonized 功能。

关于上述问题，接受的答案为我们提供了一个关于如何将其与 jupyter notebook 一起使用的示例。

但是，当我尝试使用 disutils 构建 pyx 文件时，它不起作用。

我们显然尝试使用

运行脚本

kernprof -l -v script.py

它只返回 Timer unit 耗时。

如果我尝试使用 @profile 修饰 cython 文件上的函数，它不会编译返回:

undeclared name not builtin: profile

有什么想法吗？

Answer 1

profile 装饰器由 kernprof 注入(inject)到 globals 命名空间中，因此在编译时不可用。但是，您可以 apply the profile decorator to a function even after it has been defined .例如，在您的 script.py 中，您可以编写以下内容。

from cython_module import function_to_be_profiled
# Apply the `profile` decorator
function_to_be_profiled = profile(function_to_be_profiled)

# Use function_to_be_profiled as intended

如果您使用标准 python 运行脚本，即 python script.py，代码段的第三行将失败，因为未定义 profile 装饰器。但如果您使用 kernprof 运行它，它应该会按预期运行。