提交 2c771221 authored 作者: Amjad Almahairi's avatar Amjad Almahairi

change cache code

上级 14241a7d
...@@ -216,7 +216,7 @@ class MultinomialWOReplacementFromUniform(MultinomialFromUniform): ...@@ -216,7 +216,7 @@ class MultinomialWOReplacementFromUniform(MultinomialFromUniform):
return Apply(self, [pvals, unis, as_scalar(n)], [out]) return Apply(self, [pvals, unis, as_scalar(n)], [out])
def c_code_cache_version(self): def c_code_cache_version(self):
return None return (1,)
def c_code(self, node, name, ins, outs, sub): def c_code(self, node, name, ins, outs, sub):
(pvals, unis, n) = ins (pvals, unis, n) = ins
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论