提交 5f3a30d0 authored 作者: notoraptor's avatar notoraptor

Raise an error in C code for invalid direction.

上级 42a3682e
...@@ -170,7 +170,7 @@ class BaseCorr3dMM(gof.OpenMPOp): ...@@ -170,7 +170,7 @@ class BaseCorr3dMM(gof.OpenMPOp):
def c_code_cache_version(self): def c_code_cache_version(self):
# raise this whenever modifying any of the support_code_files # raise this whenever modifying any of the support_code_files
return (6, self.openmp, blas_header_version()) return (7, self.openmp, blas_header_version())
def c_support_code_apply(self, node, nodename): def c_support_code_apply(self, node, nodename):
# REMEMBER TO RAISE c_code_cache_version when changing any of # REMEMBER TO RAISE c_code_cache_version when changing any of
...@@ -310,7 +310,10 @@ class BaseCorr3dMM(gof.OpenMPOp): ...@@ -310,7 +310,10 @@ class BaseCorr3dMM(gof.OpenMPOp):
case DIRECTION_BACKPROP_INPUTS: case DIRECTION_BACKPROP_INPUTS:
out = &%(bottom)s; out = &%(bottom)s;
break; break;
default: break; default:
PyErr_SetString(PyExc_ValueError, "CPU Corr3dMM: Invalid direction.");
{%(fail)s}
break;
} }
// Obtain or infer kernel width, height and depth // Obtain or infer kernel width, height and depth
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论