提交 e2a28991 authored 作者: Pascal Lamblin's avatar Pascal Lamblin

Change name of assert_no_cpu_op in old back-end as well

add 'assert_no_cpu_op' as a tag instead
上级 e554dd5a
......@@ -524,7 +524,8 @@ def local_assert_no_cpu_op(node):
assert_no_cpu_op = theano.tensor.opt.in2out(local_assert_no_cpu_op,
name='assert_no_cpu_op')
# 49.2 is after device specialization & fusion optimizations for last transfers
optdb.register('assert_no_cpu_op', assert_no_cpu_op, 49.2)
optdb.register('gpu_assert_no_cpu_op', assert_no_cpu_op, 49.2,
'assert_no_cpu_op')
@register_opt()
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论