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

Get on with the times.

上级 1f380740
...@@ -972,13 +972,14 @@ def local_gpua_inc_subtensor(op, context_name, inputs, outputs): ...@@ -972,13 +972,14 @@ def local_gpua_inc_subtensor(op, context_name, inputs, outputs):
@register_opt('fast_compile') @register_opt('fast_compile')
@op_lifter([tensor.AdvancedSubtensor1]) @op_lifter([tensor.AdvancedSubtensor1])
@register_opt2([tensor.AdvancedSubtensor1], 'fast_compile') @register_opt2([tensor.AdvancedSubtensor1], 'fast_compile')
def local_gpua_advanced_subtensor(op, context_name, inputs, outputs): def local_gpua_advanced_subtensor1(op, context_name, inputs, outputs):
return GpuAdvancedSubtensor1() return GpuAdvancedSubtensor1()
@register_opt('fast_compile') @register_opt('fast_compile')
@op_lifter([tensor.AdvancedSubtensor]) @op_lifter([tensor.AdvancedSubtensor])
def local_gpua_advanced_subtensor_(node, context_name): @register_opt2([tensor.AdvancedSubtensor], 'fast_compile')
def local_gpua_advanced_subtensor(op, context_name, inputs, outputs):
return GpuAdvancedSubtensor() return GpuAdvancedSubtensor()
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论