提交 d9023715 authored 作者: Ian Goodfellow's avatar Ian Goodfellow

temporarily disabled local gpugemm to prevent memory leaks

上级 c4f512af
...@@ -835,10 +835,12 @@ def local_gpu_join(node): ...@@ -835,10 +835,12 @@ def local_gpu_join(node):
return [replacement_node] return [replacement_node]
@local_optimizer([gpu_gemm_no_inplace]) #Commented out because it can result in shared = dimshuffle(gemm_inplace(dimshuffle(shared)))
def local_inplace_gemm(node): #which causes memory leaks (long term fix is to make the above not leak memory)
if node.op == gpu_gemm_no_inplace: #@local_optimizer([gpu_gemm_no_inplace])
return [gpu_gemm_inplace(*node.inputs)] #def local_inplace_gemm(node):
# if node.op == gpu_gemm_no_inplace:
# return [gpu_gemm_inplace(*node.inputs)]
# After destroyhandler is in but before we try to make elemwise things inplace # After destroyhandler is in but before we try to make elemwise things inplace
# Try to make gpu gemm inplace # Try to make gpu gemm inplace
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论