提交 bc04c904 authored 作者: Frederic Bastien's avatar Frederic Bastien

Add missing context

上级 f1a3d267
...@@ -53,7 +53,9 @@ class GPUAMultinomialFromUniform(gpuarray.basic_ops.GpuKernelBase, Op): ...@@ -53,7 +53,9 @@ class GPUAMultinomialFromUniform(gpuarray.basic_ops.GpuKernelBase, Op):
'GpuMultinomialFromUniform works only if ' 'GpuMultinomialFromUniform works only if '
'self.odtype == pvals.dtype', odtype, pvals.dtype) 'self.odtype == pvals.dtype', odtype, pvals.dtype)
br = (pvals.broadcastable[1], pvals.broadcastable[0]) br = (pvals.broadcastable[1], pvals.broadcastable[0])
out = GpuArrayType(broadcastable=br, dtype=odtype)() out = GpuArrayType(broadcastable=br,
dtype=odtype,
context_name=ctx_name)()
return Apply(self, [pvals, unis], [out]) return Apply(self, [pvals, unis], [out])
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论