跨文件使用cython.final避免虚表的正确方法

我有一个包含多个cdef类的代码库，所有类都标记为@cython.final。但由于这些类分布在不同的pyx/pxd文件中，Cython无法利用final特性避免生成虚表——因为Cython仅生成C源文件，而非C头文件。

最简示例代码 #

a.pxd #

cimport cython

@cython.final
cdef class A:
    cdef int x
    cdef int f(self)

b.pxd #

cimport cython
from a cimport A

@cython.final
cdef class B:
    cdef A a
    cdef void f(self)

a.pyx #

cimport cython

@cython.final
cdef class A:
    def __init__(self, int x):
        self.x = x
    cdef int f(self):
        return self.x

b.pyx #

cimport cython
from a cimport A

@cython.final
cdef class B:
    def __init__(self, A a):
        self.a = a
    cdef void f(self):
        self.a.f()

问题现象 #

分别用cythonize a.pyx和cythonize b.pyx编译后，B.f(A a)不会直接调用__pyx_f_1a_1A(a)，而是通过A的虚表：

$ grep "__pyx_vtab)" b.c
[...] ((struct __pyx_vtabstruct_1a_A *)__pyx_v_self->a->__pyx_vtab)->f(__pyx_v_self->a);

请问该如何正确处理？若要避免虚表，是否不能将类拆分到不同文件中？

要让Cython在跨文件场景下识别@cython.final并避免虚表调用，有以下几种可行方式：

1. 统一编译所有相关文件 #

不要单独编译每个pyx文件，而是将所有依赖的pyx文件作为一个整体传递给cythonize，比如：

cythonize a.pyx b.pyx

或者在setup.py中配置把相关文件打包成一个扩展模块。这样Cython在编译时能全局分析所有final类的定义，确认没有子类后直接生成非虚表的直接调用。

2. 在pxd文件中显式声明方法为inline #

在a.pxd中把cdef int f(self)改成cdef inline int f(self)，同时保留@cython.final标记：

# a.pxd
cimport cython

@cython.final
cdef class A:
    cdef int x
    cdef inline int f(self)

然后在a.pyx中实现这个inline方法。这样Cython在编译b.pyx时能直接获取方法的可调用信息，无需通过虚表间接调用。

3. 合并依赖类的定义（不推荐） #

如果不想调整编译方式，可将A类的定义和实现尽可能合并到单个pxd文件，让B所在文件能直接获取A的完整final特性信息。但这种方式会破坏代码模块化结构，仅适合小型项目。

关键原理 #

Cython仅在确认final类没有子类、方法不会被覆盖时，才会跳过虚表生成。跨文件单独编译时，每个文件的编译上下文独立，Cython无法知晓其他文件中的类是否真的没有被继承，因此会 fallback 到虚表调用。只有当编译上下文包含所有相关文件，或通过inline明确暴露方法的直接调用特性时，才能触发优化。

内容的提问来源于stack exchange，提问作者Bermudes