提交 53712739 authored 作者: Frederic's avatar Frederic

register local_remove_all_assert at other place where assert can be introduced

上级 edf974b4
...@@ -1598,6 +1598,12 @@ def local_remove_all_assert(node): ...@@ -1598,6 +1598,12 @@ def local_remove_all_assert(node):
compile.optdb['canonicalize'].register('local_remove_all_assert', compile.optdb['canonicalize'].register('local_remove_all_assert',
local_remove_all_assert, local_remove_all_assert,
use_db_name_as_tag=False) use_db_name_as_tag=False)
compile.optdb['stabilize'].register('local_remove_all_assert',
local_remove_all_assert,
use_db_name_as_tag=False)
compile.optdb['specialize'].register('local_remove_all_assert',
local_remove_all_assert,
use_db_name_as_tag=False)
@register_specialize @register_specialize
@gof.local_optimizer([T.Elemwise]) @gof.local_optimizer([T.Elemwise])
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论