提交 82af9d7b authored 作者: Frederic Bastien's avatar Frederic Bastien

Add an assert

上级 aa483425
......@@ -48,6 +48,7 @@ class GPUAMultinomialFromUniform(gpuarray.basic_ops.GpuKernelBase, Op):
odtype = pvals.dtype
else:
odtype = self.odtype
assert odtype == 'float32', odtype
if odtype != pvals.dtype:
raise NotImplementedError(
'GpuMultinomialFromUniform works only if '
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论