关于f2py回调的段错误

时间:2016-04-20 19:45:21

标签: python numpy fortran f2py

我正在研究分形生成器代码。主代码用python编写,迭代部分用fortran编写。我使用f2py将两个代码粘合在一起。 这是我使用的fortran函数:

function iterate(z0, func, zmax, niter) result(n)

    implicit none

    complex(kind=8), intent(in) :: z0
    real(kind=8), intent(in) :: zmax
    integer, intent(in) :: niter
    external func
    complex(kind=8) :: func

    integer :: n
    complex(kind=8) :: z

    n = 0
    z = z0
    do while ((n < niter) .and. (abs(z) < zmax))
        z = func(z)
        n = n + 1
    end do

end function iterate 

以下是生成的包装器代码的docstring:

n = iterate(z0,func,zmax,niter,[func_extra_args])

Wrapper for ``iterate``.

Parameters
----------
z0 : input complex
func : call-back function
zmax : input float
niter : input int

Other Parameters
----------------
func_extra_args : input tuple, optional
    Default: ()

Returns
-------
n : int

Notes
-----
Call-back functions::

  def func(z): return z
  Required arguments:
    z : input complex
  Return objects:
    z : complex

我在尝试时遇到Segmentation fault错误 将iterate与任何python回调函数一起使用。 以下是我得到的示例结果:

>>> from foo import iterate
>>> iterate(1.0j, lambda x: 4.0 + x**2,  4.0, 256)
Segmentation fault

我查看了有关f2py中回调的所有可用文档,但是 没有找到解决这个问题的方法。任何帮助将不胜感激。

更新

这是来自gdb的回溯:

Program received signal SIGSEGV, Segmentation fault.
cb_func_in_iterate2__user__routines (return_value=0x7fffffffdbc0, z_cb_capi=0x3ff0000000000000)
    at /tmp/tmpT8xG1q/src.linux-x86_64-2.7/juliamodule.c:470
470 /tmp/tmpT8xG1q/src.linux-x86_64-2.7/juliamodule.c: No such file or directory.
(gdb) backtrace 
#0  cb_func_in_iterate2__user__routines (return_value=0x7fffffffdbc0, z_cb_capi=0x3ff0000000000000)
    at /tmp/tmpT8xG1q/src.linux-x86_64-2.7/juliamodule.c:470
#1  0x00007ffff6b6482b in iterate2 (z0=(1,1), func=func@entry=0x7ffff6b60c20 <cb_func_in_iterate2__user__routines>, zmax=4, niter=256)
    at julia.f90:38
#2  0x00007ffff6b64897 in f2pywrapiterate2 (iterate2f2pywrap=0, z0=(1,1), func=func@entry=0x7ffff6b60c20 <cb_func_in_iterate2__user__routines>, 
    zmax=4, niter=256) at /tmp/tmpT8xG1q/src.linux-x86_64-2.7/julia-f2pywrappers.f:25
#3  0x00007ffff6b61f5e in f2py_rout_julia_iterate2 (capi_self=<optimized out>, capi_args=<optimized out>, capi_keywds=<optimized out>, 
    f2py_func=0x7ffff6b64880 <f2pywrapiterate2>) at /tmp/tmpT8xG1q/src.linux-x86_64-2.7/juliamodule.c:811
#4  0x00000000004caaa1 in PyEval_EvalFrameEx ()
#5  0x00000000004c87a1 in PyEval_EvalCodeEx ()
#6  0x00000000005030ef in ?? ()
#7  0x00000000004f8c72 in PyRun_FileExFlags ()
#8  0x00000000004f7d77 in PyRun_SimpleFileExFlags ()
#9  0x00000000004982f2 in Py_Main ()
#10 0x00007ffff6f12b45 in __libc_start_main (main=0x497d80 <main>, argc=2, argv=0x7fffffffe2a8, init=<optimized out>, fini=<optimized out>, 
    rtld_fini=<optimized out>, stack_end=0x7fffffffe298) at libc-start.c:287
#11 0x0000000000497ca0 in _start ()

1 个答案:

答案 0 :(得分:3)

我有一个解决方法,但不是一个完整的修复,我害怕。关键点似乎是f2pyfunc是一个函数时会生成一些奇怪的,可能只是不正确的c代码。将func更改为子例程似乎可以使其正常工作。

要使代码生效,请更改以下内容:

external func
complex(kind=8) :: func

...

do while ((n < niter) .and. (abs(z) < zmax))
    z = func(z)
...

interface foo
   subroutine func(f, z)
     complex(kind=8), intent(out) :: f
     complex(kind=8), intent(in) :: z
   end subroutine func
end interface foo

...

do while ((n < niter) .and. (abs(z) < zmax))
    call func(z, z)
...

这对我来说很好,并给出了正确的答案。如果你对此感到满意,你可以停止阅读。以下是我如何解决它的方法。

至于为什么会这样,你需要运行f2py之类的

f2py -m modname -h iterate.pyf iterate.f90
f2py -c --debug --build-dir build iterate.pyf iterate.f90

然后我们可以使用gdb python调试代码,然后

run -c "import iterate; iterate.iterate(1.0j, lambda x: 4.0 + x*x, 4.0, 256)"

(虽然通过它的声音,无论如何你可能已经到目前为止了)。这样做,我们在build/src.linux-x86_64.3.4/iteratemodule.c中的以下行找到了段错误(您的目录可能不同):

complex_double z=(*z_cb_capi);

z_cb_capi0x0,因此是段错误。这出现在Cython帮助函数中,当在Fortran代码中调用func时实际调用该函数。函数声明出现在上面的行之前,并且在func是函数而不是子例程的版本中,它看起来像下面的混乱:

static 
#ifdef F2PY_CB_RETURNCOMPLEX
complex_double
#else
void
#endif
 cb_func_in_iterate__user__routines (
#ifndef F2PY_CB_RETURNCOMPLEX
complex_double *return_value
#endif

#ifndef F2PY_CB_RETURNCOMPLEX
,
#endif
complex_double *z_cb_capi) {

基本上是以下之一:

static void cb_func_in_iterate__user__routines (complex_double *return_value, complex_double *z_cb_capi)
static complex_double cb_func_in_iterate__user__routines (complex_double *z_cb_capi)

取决于是否定义了F2PY_CB_RETURNCOMPLEX。似乎F2PY_CB_RETURNCOMPLEX永远不会被定义,因为gdb将该函数报告为具有第一个定义,但它被调用为好像它是第二个。您可以看到这一点,因为return_value设置为Fortran代码中z的地址,而NULL没有传递给z_cb_capi

所以,这最终会让我们找到解决此问题的第二种方法:将-DF2PY_CB_RETURNCOMPLEX传递给f2py

f2py -c -DF2PY_CB_RETURNCOMPLEX --debug --build-dir build iterate.pyf iterate.f90

然后以第二种形式编译cb_func_in_iterate__user__routines,这意味着它会被正确调用。

现在,我认为你不应该这样做,所以我怀疑这是f2py中的一个错误。