在C中使用结构(可能与LAPACKE相关)中的指针时出现段错误



由于这是我在这里的第一篇文章,我希望我不会犯太多错误。

我正面临着以下问题。当我运行这段代码(相当长,我很抱歉)时,我在LAPACKE调用期间的"解决"函数中出现了分段错误。

问题是,当我不试图在main函数中打开文件"dataf"时,此代码运行时没有任何(可见)问题。

我想我把类型Bob和指针弄乱了,但我想听听你的意见。

问好鲍特

#include <stdio.h>
#include <stdlib.h>
#include <lapacke/lapacke.h>
typedef struct{
   double *data;
   double *tmp;
   double **A;
   int size;
 } Bob;
void allocate(Bob *b);
void init_data(Bob *b);
void init_A(Bob* b);
void allocate(Bob *b){
   b->data = malloc(b->size*sizeof(double));
   b->tmp = malloc(b->size*sizeof(double));
   b->A = malloc(4*sizeof(double*));
   b->A[0] = malloc(b->size*sizeof(double));
   b->A[1] = malloc(b->size*sizeof(double));
   b->A[2] = malloc(b->size*sizeof(double));
   b->A[3] = malloc(b->size*sizeof(double));
}
void init_data(Bob *b){
   int i;
   for (i=0;i<b->size;i++){
      b->data[i] = 0.0;
   }
}
void init_A(Bob* b){
   int i;
   double *diag = b->A[0];
   double *down = b->A[1];
   double *up = b->A[2];
   double *up2 = b->A[3];
   for (i=0;i<b->size;i++){
     diag[i] = 2.0;
     down[i] = -1.0;
     up[i] = -1.0;
     up2[i] = 0.0;
   }
}
void solve(Bob *b){
   lapack_int info;
   lapack_int *ipiv = malloc(b->size*(sizeof(lapack_int)));
   lapack_int lsize = b->size;
   lapack_int lone = 1;
   int i;
   for (i=0;i<b->size;i++){
      b->tmp[i] = b->data[i];
   }
   info = LAPACKE_dgttrs(LAPACK_COL_MAJOR,'N',
     lsize,lone,
     b->A[1],b->A[0],b->A[2],b->A[3],
     ipiv, b->tmp, lsize);
   printf("%dn",info);
   free(ipiv);
 }
int main(void){
   Bob b;
   b.size = 10000;
   allocate(&b);
   init_A(&b);
   init_data(&b);
// No Segfault if dataf is not opened
   FILE *dataf;
   dataf = fopen("data","w");
   fclose(dataf);
   solve(&b);
   free(b.data);
   free(b.tmp);
   free(b.A[0]);
   free(b.A[1]);
   free(b.A[2]);
   free(b.A[3]);
   free(b.A);
   return 0;
}
与Makefile:

P=mini
CC=gcc
CFLAGS=-g -Wall
LFLAGS=-lm -llapacke -llapack -lblas
FILES=min.c
all: $(FILES)
   $(CC) $(CFLAGS) $(LFLAGS) $(FILES) -o $(P)

编辑:我在原来的帖子中犯了一个错误。A[0] -> A[3]的赋值类型是双精度体,而不是指向双精度体的指针。

在解决后改变fclose的位置似乎可以解决问题。但在我的实际代码中,它没有改变任何东西……

EDIT2:问题似乎与LAPACK功能有关。我改变了这个函数(更适合我的问题),内存问题现在消失了。

这不是我原来问题的解决方案,但我用某种方法解决了它。谢谢所有的回复和评论,无论如何都是非常有用的:)

EDIT3:我找到了问题的根源。它与LAPACK有关。在函数求解中,我为ipiv分配空间。但是lapacke_dgtrs接受init_A中的lapacke_dgtrf函数给出的ipiv作为输入。

所以我在结构体中为ipiv分配了一个指针,并在程序的其余部分使用它

由于fclose函数导致的错误,在fclose之前检查数据是否为NULL或noy

FILE *dataf;
dataf = fopen("data","w"); //If file is not opened for writing, the dataf is NULL
fclose(dataf);   
solve(&b);

改为

FILE *dataf;
dataf = fopen("data","w");
if(dataf != NULL)
    fclose(dataf);
solve(&b);

在你对"Bob"的定义中,成员A被定义为指向指针的指针,根据你的代码,我假设你正在设置一个二维双精度数组。如果是这种情况,分配a b-> a [i]的四个语句应该分配双精度,而不是指向双精度的指针。

void allocate(Bob *b){
   b->data = malloc(b->size*sizeof(double));
   b->tmp = malloc(b->size*sizeof(double));
   b->A = malloc(4*sizeof(double*));
   b->A[0] = malloc(b->size*sizeof(double));
   b->A[1] = malloc(b->size*sizeof(double));
   b->A[2] = malloc(b->size*sizeof(double));
   b->A[3] = malloc(b->size*sizeof(double));
}

相关内容

  • 没有找到相关文章

最新更新