提交 461f2cdd authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Add the inplace opt.

上级 4b10d6f2
......@@ -533,3 +533,12 @@ def local_scan_to_gpua(node):
nw_op = scan_op.Scan(scan_ins, scan_outs, info,
typeConstructor=GpuArrayType).make_node(*nw_ins)
return nw_op.outputs
optdb.register('gpua_scanOp_make_inplace',
scan_opt.ScanInplaceOptimizer(typeConstructor=GpuArrayType,
gpu_flag=True),
75,
'gpua',
'fast_run',
'inplace',
'scan')
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论