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

fix new crash in gpu optimiztion due to the new scheduler stuff.

上级 74147ec4
...@@ -1471,7 +1471,7 @@ def gpuScanOptimization(node): ...@@ -1471,7 +1471,7 @@ def gpuScanOptimization(node):
tmp_in, tmp_out = gpu_reconstruct_graph(scan_ins, tmp_in, tmp_out = gpu_reconstruct_graph(scan_ins,
scan_outs) scan_outs)
local_fgraph = gof.FunctionGraph(tmp_in, tmp_out) local_fgraph = gof.FunctionGraph(tmp_in, tmp_out)
_cmodule_key = gof.CLinker.cmodule_key_(local_fgraph, []) _cmodule_key = gof.CLinker().cmodule_key_(local_fgraph, [])
info['gpu_hash'] = hash(_cmodule_key) info['gpu_hash'] = hash(_cmodule_key)
typeConstructor = lambda broadcastable, dtype: CudaNdarrayType( typeConstructor = lambda broadcastable, dtype: CudaNdarrayType(
...@@ -1520,7 +1520,7 @@ def gpuScanOptimization(node): ...@@ -1520,7 +1520,7 @@ def gpuScanOptimization(node):
tmp_in, tmp_out = gpu_reconstruct_graph(scan_ins, tmp_in, tmp_out = gpu_reconstruct_graph(scan_ins,
scan_outs) scan_outs)
local_fgraph = gof.FunctionGraph(tmp_in, tmp_out) local_fgraph = gof.FunctionGraph(tmp_in, tmp_out)
_cmodule_key = gof.CLinker.cmodule_key_(local_fgraph, []) _cmodule_key = gof.CLinker().cmodule_key_(local_fgraph, [])
info['gpu_hash'] = hash(_cmodule_key) info['gpu_hash'] = hash(_cmodule_key)
def typeConstructor(broadcastable, dtype): def typeConstructor(broadcastable, dtype):
assert dtype == 'float32' assert dtype == 'float32'
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论