提交 a75fe6cb authored 作者: affanv14's avatar affanv14

add extra check for gpucorr3dmm

上级 9ef76c3f
...@@ -485,6 +485,11 @@ PyGpuArrayObject* corr3dMM(PyGpuArrayObject *const bottom, ...@@ -485,6 +485,11 @@ PyGpuArrayObject* corr3dMM(PyGpuArrayObject *const bottom,
"GpuCorr3dMM images and kernel must have the same stack size\n"); "GpuCorr3dMM images and kernel must have the same stack size\n");
return NULL; return NULL;
} }
if ((nFilters % numgroups) != 0) {
PyErr_SetString(PyExc_ValueError,
"CorrMM the number of filters must be divisible by the number of groups\n");
return NULL;
}
// implicit dilated filter // implicit dilated filter
const size_t dil_kH = (kH - 1) * dilH + 1; const size_t dil_kH = (kH - 1) * dilH + 1;
const size_t dil_kW = (kW - 1) * dilW + 1; const size_t dil_kW = (kW - 1) * dilW + 1;
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论