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

Question

我对 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 之前停止，有时在最后的循环内停止，有时根本不停止。停止的概率取决于 lakeNr 和 TrialNr 的值，尽管我没有发现任何有用的相关性。这可能只是由于输入矩阵的维度发生变化时填充的随机数不同所致。 在任何情况下都不会抛出异常。程序就停止了，就好像它已经完成一样。

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

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;
}

带有 double 的版本工作正常并且不会引起任何问题。我猜这个问题与内存管理有关，但我不确定。有谁知道 int 版本的问题可能是什么？

我正在使用 python 3.4 64 位和 Windows 8 64 位(编译器:Visual Studio 10)。

感谢您的帮助!

Answer 1

我想出了如何避免这个问题:为输出数组分配内存的函数ptrvectorInt无法正常工作。我把它替换为

int **ptrvectorInt(long dim1) {
   int **v;
   if (!(v = malloc(dim1 * sizeof(int*)))) {
      PyErr_SetString(PyExc_MemoryError,
              "In **ptrvectorInt. Allocation of memory for integer array failed.");
      exit(0); 
   }
   return v;
}

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