提交 53e6d2c3 authored 作者: Frederic's avatar Frederic

pep8

上级 052da844
...@@ -1593,7 +1593,7 @@ def gpuScanOptimization(node): ...@@ -1593,7 +1593,7 @@ def gpuScanOptimization(node):
# __init__ does not know about cuda ndarray and can not # __init__ does not know about cuda ndarray and can not
# handle graphs with inputs being Cuda Ndarrays # handle graphs with inputs being Cuda Ndarrays
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)
...@@ -1640,7 +1640,7 @@ def gpuScanOptimization(node): ...@@ -1640,7 +1640,7 @@ def gpuScanOptimization(node):
# __init__ does not know about cuda ndarray and can not # __init__ does not know about cuda ndarray and can not
# handle graphs with inputs being Cuda Ndarrays # handle graphs with inputs being Cuda Ndarrays
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)
...@@ -1662,7 +1662,7 @@ def gpuScanOptimization(node): ...@@ -1662,7 +1662,7 @@ def gpuScanOptimization(node):
optdb.register('gpu_scanOp_make_inplace', optdb.register('gpu_scanOp_make_inplace',
scan_opt.ScanInplaceOptimizer(typeConstructor=typeConstructor, scan_opt.ScanInplaceOptimizer(typeConstructor=typeConstructor,
gpu_flag=True), gpu_flag=True),
75, 75,
'gpu', 'gpu',
'fast_run', 'fast_run',
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论