C扩展中的函数随机停止python程序执行

时间:2015-05-21 23:59:57

标签: python c memory-management random python-c-extension

我对python的C扩展相对较新。我写了一个扩展,显示了一个对我来说很奇怪的行为。当我使用此扩展程序运行python脚本时,脚本会在中成功执行扩展程序中的后随机停止。也就是说,我有一个类似的脚本:

import FlowCalc
import numpy as np
np.random.seed(1)

lakeNr = 6
trialNr = 10
a = np.round(np.random.rand(trialNr, lakeNr)).astype(int)
b = np.ones(shape=(lakeNr, lakeNr), dtype=float)

x = FlowCalc.flowCalc(a, b)
print(x)

for i in range(100000):
    print(i)

脚本有时会在打印x之前停止,有时会在循环结束时停止,有时它根本不会停止。停止的概率取决于lakeNrtrialNr的值,但我没有发现任何有用的相关性。这可能是因为输入矩阵的尺寸变化时填充输入矩阵的随机数不同。 在任何情况下都不会抛出异常。该程序就像它完成一样停止。

我能够检测到我的扩展中必须对此行为负责的函数。首先,我向您展示我的包装函数:

static PyObject *FlowCalc_flowCalc(PyObject *self, PyObject *args)
{
    PyArrayObject *trials_array, *flows_array, *result;

    /* Parse the input tuple */
    if (!PyArg_ParseTuple(args, "OO", &trials_array, &flows_array)) {
        PyErr_SetString(PyExc_ValueError,
                    "Exception");
        return NULL;
    }
    pymatrix_to_CarrayptrsInt(trials_array);

    return Py_BuildValue("i", 42);

问题必须放在函数pymatrix_to_CarrayptrsInt

int **pymatrix_to_CarrayptrsInt(PyArrayObject *arrayin) {
   int **c, *a;
   int i,n,m;

   n=arrayin->dimensions[0];
   m=arrayin->dimensions[1];
   c=ptrvectorInt(n);
   a=(int *) arrayin->data; /* pointer to arrayin data as int */
   for ( i=0; i<n; i++) {
      c[i]=a+i*m; }
   return c;
}

int **ptrvectorInt(long n) {
   int **v;
   v = (int**) malloc((size_t) (n * sizeof(int)));
   if (!v)   {
      printf("In **ptrvectorInt. Allocation of memory for int array failed.");
      exit(0); }
   return v;
}

此方法是pymatrix_to_CarrayptrsDouble

的更改重新实现
double **pymatrix_to_CarrayptrsDouble(PyArrayObject *arrayin) {
   double **c, *a;
   int i,n,m;

   n=arrayin->dimensions[0];
   m=arrayin->dimensions[1];
   c=ptrvectorDouble(n);
   a=(double *) arrayin->data; /* pointer to arrayin data as double */
   for ( i=0; i<n; i++) {
      c[i]=a+i*m; }
   return c;
}

double **ptrvectorDouble(long n) {
   double **v;
   v = (double**) malloc((size_t) (n * sizeof(double)));
   if (!v)   {
      printf("In **ptrvectorDouble. Allocation of memory for double array failed.");
      exit(0); }
   return v;
}

双版本工作正常,不会造成任何问题。我猜这个问题与内存管理有关,但我不确定。有没有人知道int版本的问题是什么?

使用python 3.4 64bit和Windows 8 64bit(编译器:Visual Studio 10)。

感谢您的帮助!

1 个答案:

答案 0 :(得分:0)

我想出了如何避免这个问题:为输出数组分配内存的函数Me.dateA = Me.dateA.AddMonths(1) lblDateA.Text = Me.dateA.ToShortDateString() 无法正常工作。我用

替换了它
ptrvectorInt

一切正常。我仍然不完全知道错误的机制(即为什么它出现以及为什么它随机停止程序)。但问题已经解决了。