提交 596e3cdb authored 作者: lamblin's avatar lamblin

Merge pull request #968 from nouiz/fix_crash_verbose

Fix crash in gpu code in verbose mode when after using fprint_CudaNdarra...
...@@ -4501,24 +4501,31 @@ void fprint_CudaNdarray(FILE * fd, const CudaNdarray *self) ...@@ -4501,24 +4501,31 @@ void fprint_CudaNdarray(FILE * fd, const CudaNdarray *self)
fprintf(fd, "%i\t", CudaNdarray_HOST_STRIDES(self)[i]); fprintf(fd, "%i\t", CudaNdarray_HOST_STRIDES(self)[i]);
} }
int data=0; if (self->dev_structure)
fprintf(fd, "\n\tDEV_DIMS: ");
for (int i = 0; i < self->nd; ++i)
{ {
cublasGetVector(1, sizeof(int), int data=0;
self->dev_structure+i, 1, fprintf(fd, "\n\tDEV_DIMS: ");
&data, 1); for (int i = 0; i < self->nd; ++i)
fprintf(fd, "%i\t", data); {
cublasGetVector(1, sizeof(int),
self->dev_structure+i, 1,
&data, 1);
fprintf(fd, "%i\t", data);
}
fprintf(fd, "\n\tDEV_STRIDES: ");
for (int i = 0; i < self->nd; ++i)
{
cublasGetVector(1, sizeof(int),
self->dev_structure + self->nd+i, 1,
&data, 1);
fprintf(fd, "%i \t", data);
}
fprintf(fd, "\n");
} }
fprintf(fd, "\n\tDEV_STRIDES: "); else
for (int i = 0; i < self->nd; ++i)
{ {
cublasGetVector(1, sizeof(int), fprintf(fd, "\n\tdev_structure not allocated\n");
self->dev_structure + self->nd+i, 1,
&data, 1);
fprintf(fd, "%i \t", data);
} }
fprintf(fd, "\n");
} }
/* /*
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论