提交 099efddb authored 作者: James Bergstra's avatar James Bergstra

commenting some debugprints in blas.py

上级 800b252d
...@@ -889,6 +889,7 @@ def _gemm_from_node2(node): ...@@ -889,6 +889,7 @@ def _gemm_from_node2(node):
if len(lst) > 1: if len(lst) > 1:
lst = _factor_canonicalized(lst) lst = _factor_canonicalized(lst)
rval = _gemm_from_factored_list(lst) rval = _gemm_from_factored_list(lst)
#print "RVAL", rval
if rval: if rval:
assert rval[0].type == node.outputs[0].type, (rval[0].type, node.outputs[0].type) assert rval[0].type == node.outputs[0].type, (rval[0].type, node.outputs[0].type)
return rval return rval
...@@ -909,7 +910,6 @@ class GemmOptimizer(Optimizer): ...@@ -909,7 +910,6 @@ class GemmOptimizer(Optimizer):
did_something = False did_something = False
nodelist.reverse() nodelist.reverse()
for node in nodelist: for node in nodelist:
#new_outputs = _gemm_from_node(node)
try: try:
new_outputs = _gemm_from_node2(node) new_outputs = _gemm_from_node2(node)
except InconsistencyError, e: except InconsistencyError, e:
...@@ -1193,9 +1193,10 @@ blas_optdb.register('local_dot22_to_dot22scalar', ...@@ -1193,9 +1193,10 @@ blas_optdb.register('local_dot22_to_dot22scalar',
11, 'fast_run') 11, 'fast_run')
from opt import register_specialize from opt import register_specialize, register_canonicalize
#@register_specialize #@register_specialize
@local_optimizer([]) @local_optimizer([])
def local_print_as_we_go_along(node): def local_print_as_we_go_along(node):
if node.op in (T.sub, T.add): if node.op in (T.sub, T.add):
debugprint(node) debugprint(node)
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论