Cython最终扩展类型与结构体的内存分配及参数传递问题

一、基础问题 #

问题：Cython是否会对最终扩展类型进行栈分配？是否会在其他扩展类型中内联分配这类类型？C结构体的情况又如何？ #

• 最终扩展类型（标记@cython.final的cdef class）：
  • 不会进行栈分配，所有扩展类型对象均在堆上创建；
  • 不会在其他扩展类型中内联分配，扩展类型中的成员本质是指向堆对象的指针，而非直接嵌入对象内存。
• C结构体：
  • 支持栈分配：在函数内通过cdef struct_name var声明时，结构体实例会分配在栈上；
  • 支持内联分配：当结构体作为其他结构体或扩展类型的成员时，会直接嵌入到宿主的内存空间中，而非以指针形式存在。

二、基于示例代码的问题 #

示例代码：

import cython

@cython.final
cdef class Fish:
   cdef double weight

cdef class Aquarium:
   cdef Fish fish

cpdef double fish_weight(Fish fish):
   return fish.weight

cpdef void change_fish_weight(Fish fish, double weight):
   fish.weight = weight

cpdef double new(double weight):
   cdef Fish f = Fish()
   f.weight = weight
   return f.weight


cdef Aquarium aq = Aquarium()
cdef Fish fish = Fish()

change_fish_weight(fish, 5.0)
fish_weight(fish)
new(5.0)

1. Fish对象会在Aquarium对象中被内联分配吗？ #

不会。Aquarium中的fish成员是指向Fish对象的指针，Fish对象本身在堆上独立分配，不会嵌入到Aquarium的内存空间里。

2. fish会以「引用传递」的方式传入change_fish_weight函数吗？ #

是。Cython中扩展类型对象作为参数传递时，本质是传递对象的指针（即引用传递），函数内对对象成员的修改会直接作用于原对象，这里change_fish_weight修改fish.weight会改变外部传入的fish对象的值，符合引用传递的特征。

3. fish会以「引用传递」的方式传入fish_weight函数吗？ #

是。即便函数仅读取对象成员，扩展类型对象参数依然按引用传递（传递指针），而非值拷贝。因为扩展类型对象在堆上，传递指针比拷贝整个对象更高效，Cython默认采用这种方式。

4. 在new函数中，f会被分配在栈上吗？（假设优化器不会完全消除不必要的变量f） #

不会。Fish是扩展类型，无论是否标记final，对象都会在堆上分配，cdef Fish f = Fish()实际是创建堆对象后，让f持有指向该对象的指针，f本身是栈上的指针变量，但Fish对象的内存在堆上。

5. 如果Fish是结构体而非扩展类型，上述第4题的答案会有什么变化？ #

如果将Fish改为C结构体（cdef struct Fish: double weight），那么new函数中的cdef Fish f = Fish()会将f分配在栈上（在优化器不消除变量的前提下）。结构体是值类型，默认在栈上分配实例。

三、补充问题 #

问题：启用编译器优化后，哪些问题的答案会发生变化？原因是什么？ #

• 可能变化的场景：
  • 问题4：若没有“优化器不消除变量”的假设，编译器（如GCC/Clang）在-O2/-O3级别优化下，可能会完全消除变量f，直接将weight赋值后返回，此时不存在栈或堆分配的操作；但题目明确假设优化器不消除变量，所以答案不变。
  • 问题5的结构体场景：优化器可能会做激进的语义保留优化，比如将结构体的操作直接展开，甚至完全省略栈分配的结构体，但同样在题目假设不消除变量的前提下，栈分配的结论不变。
• 不会变化的场景：基础问题中的扩展类型分配规则、代码问题1-3的参数传递方式，这些是Cython的语言特性决定的，编译器优化不会改变这些核心规则，只会在不改变语义的前提下减少冗余操作。
• 核心原因：编译器优化的目标是在保证程序语义不变的前提下，通过消除冗余变量、简化内存操作、展开计算等方式提升性能，但不会改变Cython对扩展类型、结构体的内存分配与参数传递的基础规则。

内容的提问来源于stack exchange，提问作者Přemysl Šťastný