提交 47738f11 authored 作者: Frederic's avatar Frederic

[CRASH] fix new compilation crash introduced.

上级 a53b72fb
...@@ -249,12 +249,12 @@ class GpuMultinomialFromUniform(MultinomialFromUniform, GpuOp): ...@@ -249,12 +249,12 @@ class GpuMultinomialFromUniform(MultinomialFromUniform, GpuOp):
fail = sub['fail'] fail = sub['fail']
return """ return """
if (PyArray_NDIM(%(pvals)s) != 2) if (CudaNdarray_NDIM(%(pvals)s) != 2)
{ {
PyErr_Format(PyExc_TypeError, "pvals wrong rank"); PyErr_Format(PyExc_TypeError, "pvals wrong rank");
%(fail)s; %(fail)s;
} }
if (PyArray_NDIM(%(unis)s) != 1) if (CudaNdarray_NDIM(%(unis)s) != 1)
{ {
PyErr_Format(PyExc_TypeError, "unis wrong rank"); PyErr_Format(PyExc_TypeError, "unis wrong rank");
%(fail)s; %(fail)s;
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论