提交 65eb53a5 authored 作者: --global's avatar --global

Only extract chosen algo after checking for success

上级 07931d20
...@@ -83,8 +83,14 @@ APPLY_SPECIFIC(conv_fwd)(CudaNdarray *input, CudaNdarray *kerns, ...@@ -83,8 +83,14 @@ APPLY_SPECIFIC(conv_fwd)(CudaNdarray *input, CudaNdarray *kerns,
requestedCount, requestedCount,
&count, &count,
&choosen_algo_perf); &choosen_algo_perf);
if (err != CUDNN_STATUS_SUCCESS) {
PyErr_Format(PyExc_RuntimeError,
"GpuDnnConv: error selecting convolution algo: %s",
cudnnGetErrorString(err));
return 1;
}
chosen_algo = choosen_algo_perf.algo; chosen_algo = choosen_algo_perf.algo;
fprintf(stdout, "Choose algo %i\n", chosen_algo);
} }
else else
{ {
...@@ -97,13 +103,13 @@ APPLY_SPECIFIC(conv_fwd)(CudaNdarray *input, CudaNdarray *kerns, ...@@ -97,13 +103,13 @@ APPLY_SPECIFIC(conv_fwd)(CudaNdarray *input, CudaNdarray *kerns,
CUDNN_CONVOLUTION_FWD_SPECIFY_WORKSPACE_LIMIT, CUDNN_CONVOLUTION_FWD_SPECIFY_WORKSPACE_LIMIT,
free, free,
&chosen_algo); &chosen_algo);
}
if (err != CUDNN_STATUS_SUCCESS) { if (err != CUDNN_STATUS_SUCCESS) {
PyErr_Format(PyExc_RuntimeError, PyErr_Format(PyExc_RuntimeError,
"GpuDnnConv: error selecting convolution algo: %s", "GpuDnnConv: error selecting convolution algo: %s",
cudnnGetErrorString(err)); cudnnGetErrorString(err));
return 1; return 1;
}
} }
// Store the shapes of the inputs and kernels as well as the chosen // Store the shapes of the inputs and kernels as well as the chosen
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论