提交 4c6b567e authored 作者: James Bergstra's avatar James Bergstra 提交者: Frederic

CBLAS removing debug prints

上级 29b0fa88
...@@ -436,17 +436,14 @@ class CGemv(BaseBLAS, Gemv): ...@@ -436,17 +436,14 @@ class CGemv(BaseBLAS, Gemv):
@local_optimizer([gemv_inplace, gemv_no_inplace]) @local_optimizer([gemv_inplace, gemv_no_inplace])
def use_c_gemv(node): def use_c_gemv(node):
if node.op == gemv_no_inplace: if node.op == gemv_no_inplace:
print "inserting C_GEMV"
return [CGemv(inplace=False)(*node.inputs)] return [CGemv(inplace=False)(*node.inputs)]
if node.op == gemv_inplace: if node.op == gemv_inplace:
print "inserting dstruc C_GEMV"
return [CGemv(inplace=True)(*node.inputs)] return [CGemv(inplace=True)(*node.inputs)]
@local_optimizer([CGemv(inplace=False)]) @local_optimizer([CGemv(inplace=False)])
def make_c_gemv_destructive(node): def make_c_gemv_destructive(node):
if node.op == CGemv(inplace=False): if node.op == CGemv(inplace=False):
print "inserting destructive C_GER"
return [CGemv(inplace=True)(*node.inputs)] return [CGemv(inplace=True)(*node.inputs)]
...@@ -462,8 +459,8 @@ blas_optdb.register('use_c_blas', ...@@ -462,8 +459,8 @@ blas_optdb.register('use_c_blas',
], ],
max_use_ratio=5), max_use_ratio=5),
20, 'fast_run', 'c_blas') 20, 'fast_run', 'c_blas')
print 'BLAS_OPTDB' #print 'BLAS_OPTDB'
print blas_optdb #print blas_optdb
# this matches the InplaceBlasOpt defined in blas.py # this matches the InplaceBlasOpt defined in blas.py
optdb.register('c_blas_destructive', optdb.register('c_blas_destructive',
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论