Cython:创建一个数组抛出“不允许在一个常量表达式”
我尝试重写一个复杂的函数形式Python到Cython以加速它的大规模,我遇到以下问题:编译我的函数hh_vers_vector.pyx使用Cython:创建一个数组抛出“不允许在一个常量表达式”
setup(
ext_modules=cythonize("hh_vers_vector.pyx"),
)
它引发以下错误
cdef int numSamples = len(Iext);
# initial values
cdef float v[numSamples]
^
------------------------------------------------------------
hh_vers_vector.pyx:47:27: Not allowed in a constant expression
但是,如果我给你“NUMSAMPLES”作为一个数字的功能,是没有问题的。我不明白,因为我认为len(Iext)也会返回一个数字为10.000。那么为什么Cython对这个表达式有问题?
cdef float v[numSamples] # ERROR
cdef float v[10000] # NO ERROR
我的功能看起来像这样至今:
from math import exp
import time
def hhModel(*params, Iext, float dt, int Vref):
## Unwrap params argument: these variables are going to be optimized
cdef float ENa = params[0]
cdef float EK = params[1]
cdef float EL = params[2]
cdef float GNa = params[3]
cdef float GK = params[4]
cdef float GL = params[5]
## Input paramters
# I : a list containing external current steps, your stimulus vector [nA/1000]
# dt : a crazy time parameter [ms]
# Vref : reference potential [mV]
# T : Total simulation time in [ms]
def alphaM(float v, float vr): return 0.1 * (v-vr-25)/(1 - exp(-(v-vr-25)/10))
def betaM(float v, float vr): return 4 * exp(-(v-vr)/18)
def alphaH(float v, float vr): return 0.07 * exp(-(v-vr)/20)
def betaH(float v, float vr): return 1/(1 + exp(-(v-vr-30)/10))
def alphaN(float v, float vr): return 0.01 * (v-vr-10)/(1 - exp(-(v-vr-10)/10))
def betaN(float v, float vr): return 0.125 * exp(-(v-vr)/80)
## steady-state values and time constants of m,h,n
def m_infty(float v, float vr): return alphaM(v,vr)/(alphaM(v,vr) + betaM(v,vr))
def h_infty(float v, float vr): return alphaH(v,vr)/(alphaH(v,vr) + betaH(v,vr))
def n_infty(float v, float vr): return alphaN(v,vr)/(alphaN(v,vr) + betaN(v,vr))
## parameters
cdef float Cm, gK, gL, INa, IK, IL, dv_dt, dm_dt, dh_dt, dn_dt, aM, bM, aH, bH, aN, bN
cdef float Smemb = 4000 # [um^2] surface area of the membrane
cdef float Cmemb = 1 # [uF/cm^2] membrane capacitance density
Cm = Cmemb * Smemb * 1e-8 # [uF] membrane capacitance
gNa = GNa * Smemb * 1e-8 # Na conductance [mS]
gK = GK * Smemb * 1e-8 # K conductance [mS]
gL = GL * Smemb * 1e-8 # leak conductance [mS]
######### HERE IS THE PROBLEM ##############
cdef int numSamples = len(Iext);
######### HERE IS THE PROBLEM ##############
# initial values
cdef float v[numSamples]
cdef float m[numSamples]
cdef float h[numSamples]
cdef float n[numSamples]
v[0] = Vref # initial membrane potential
m[0] = m_infty(v[0], Vref) # initial m
h[0] = h_infty(v[0], Vref) # initial h
n[0] = n_infty(v[0], Vref) # initial n
## calculate membrane response step-by-step
for j in range(0, numSamples-1):
# ionic currents: g[mS] * V[mV] = I[uA]
INa = gNa * m[j]*m[j]*m[j] * h[j] * (ENa-v[j])
IK = gK * n[j]*n[j]*n[j]*n[j] * (EK-v[j])
IL = gL * (EL-v[j])
# derivatives
# I[uA]/C[uF] * dt[ms] = dv[mV]
dv_dt = (INa + IK + IL + Iext[j]*1e-3)/Cm;
aM = 0.1 * (v[j]-Vref-25)/(1 - exp(-(v[j]-Vref-25)/10))
bM = 4 * exp(-(v[j]-Vref)/18)
aH = 0.07 * exp(-(v[j]-Vref)/20)
bH = 1/(1 + exp(-(v[j]-Vref-30)/10))
aN = 0.01 * (v[j]-Vref-10)/(1 - exp(-(v[j]-Vref-10)/10))
bN = 0.125 * exp(-(v[j]-Vref)/80)
dm_dt = (1-m[j])* aM - m[j]*bM
dh_dt = (1-h[j])* aH - h[j]*bH
dn_dt = (1-n[j])* aN - n[j]*bN
# calculate next step
v[j+1] = (v[j] + dv_dt * dt)
m[j+1] = (m[j] + dm_dt * dt)
h[j+1] = (h[j] + dh_dt * dt)
n[j+1] = (n[j] + dn_dt * dt)
return v
cdef
在用Cython是对C的定义,顾名思义。因此,代码
cdef float v[10000]
被转换为下面的C代码
float v[10000];
含义大小10K的静态 float数组,在编译时定义。
此,在另一方面
cdef float v[numSamples]
将转化成C代码
int numSamples = <..>;
float v[numSamples];
这是编译包含可变数目的元件的静态数组,这是无效的尝试C.因此,“常量表达式错误”。
要么使用python列表来存储浮点数,要么使用dynamically allocate C arrays via malloc
/free
。
感谢您的评论!我猜pythonl列表会比C数组慢吗?我以前从来没有用过malloc也没有free,你知道代码会如何看起来像吗? –
顺便说一句,使用'DEF numSamples = 200000'也可以,但是我不明白你的解释了。 –
静态数组在编译时不能分配一个可变数字。它们是静态的。像上面这样静态定义的数字是静态/常量。在运行时更改的int变量不是。 – danny
[Cython“不允许在常量表达式中”,boundscheck False不起作用]的可能重复(https://*.com/questions/46306737/cython-not-allowed-in-a-constant-expression -boundscheck-false-doesnt-work) – DavidW
我看到那篇文章,但它并没有真正帮助.. –
我认为@ danny对这个问题的回答有点更好解释,但它的确说大体上是相同的东西(所以我我不知道为什么这个答案有帮助,另一个职位没有......)。使用另一个问题中给出的内存视图的建议是最明智的解决方案(比malloc/free更容易) – DavidW