提交 f7c01d53 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Add fast_compile for the lifter.

上级 38199e38
...@@ -485,7 +485,7 @@ def local_gpua_multinomial(op, context_name, inputs, outputs): ...@@ -485,7 +485,7 @@ def local_gpua_multinomial(op, context_name, inputs, outputs):
gpu_op(p, u)) gpu_op(p, u))
@register_opt() @register_opt('fast_compile')
@op_lifter([theano.sandbox.multinomial.MultinomialWOReplacementFromUniform]) @op_lifter([theano.sandbox.multinomial.MultinomialWOReplacementFromUniform])
@register_opt2([theano.sandbox.multinomial.MultinomialWOReplacementFromUniform], 'fast_compile') @register_opt2([theano.sandbox.multinomial.MultinomialWOReplacementFromUniform], 'fast_compile')
def local_gpua_multinomial_wor(op, context_name, inputs, outputs): def local_gpua_multinomial_wor(op, context_name, inputs, outputs):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论