提交 cb040b2f authored 作者: John Arevalo's avatar John Arevalo

Fix AttributeError: 'builtin_function_or_method'

compile is not correctly referenced and fails with "AttributeError: 'builtin_function_or_method' object has no attribute 'optdb'"
上级 565650e4
...@@ -501,7 +501,7 @@ def local_assert_no_cpu_op(node): ...@@ -501,7 +501,7 @@ def local_assert_no_cpu_op(node):
assert_no_cpu_op = theano.tensor.opt.in2out(local_assert_no_cpu_op, assert_no_cpu_op = theano.tensor.opt.in2out(local_assert_no_cpu_op,
name='assert_no_cpu_op') name='assert_no_cpu_op')
# 49.2 is after device specialization & fusion optimizations for last transfers # 49.2 is after device specialization & fusion optimizations for last transfers
theano.compile.optdb.register('assert_no_cpu_op', assert_no_cpu_op, 49.2) optdb.register('assert_no_cpu_op', assert_no_cpu_op, 49.2)
@register_opt() @register_opt()
...@@ -2575,7 +2575,7 @@ def local_inplace_gpu_sparse_block_gemv(node): ...@@ -2575,7 +2575,7 @@ def local_inplace_gpu_sparse_block_gemv(node):
new_node = gpu_sparse_block_gemv_inplace(*node.inputs) new_node = gpu_sparse_block_gemv_inplace(*node.inputs)
return [new_node] return [new_node]
return False return False
compile.optdb.register('local_inplace_gpu_sparse_block_gemv', optdb.register('local_inplace_gpu_sparse_block_gemv',
TopoOptimizer( TopoOptimizer(
local_inplace_gpu_sparse_block_gemv, local_inplace_gpu_sparse_block_gemv,
failure_callback=TopoOptimizer.warn_inplace), failure_callback=TopoOptimizer.warn_inplace),
...@@ -2591,7 +2591,7 @@ def local_inplace_gpu_sparse_block_outer(node): ...@@ -2591,7 +2591,7 @@ def local_inplace_gpu_sparse_block_outer(node):
new_node = gpu_sparse_block_outer_inplace(*node.inputs) new_node = gpu_sparse_block_outer_inplace(*node.inputs)
return [new_node] return [new_node]
return False return False
compile.optdb.register('local_inplace_gpu_sparse_block_outer', optdb.register('local_inplace_gpu_sparse_block_outer',
TopoOptimizer( TopoOptimizer(
local_inplace_gpu_sparse_block_outer, local_inplace_gpu_sparse_block_outer,
failure_callback=TopoOptimizer.warn_inplace), failure_callback=TopoOptimizer.warn_inplace),
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论