提交 3c63af3e authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Rename optimization to gpua_ to avoid conflicts with the cuda module.

上级 a8fe1069
......@@ -167,7 +167,7 @@ def max_inputs_to_GpuElemwise(node):
gpu_local_elemwise_fusion = tensor.opt.local_elemwise_fusion_op(
GpuElemwise,
max_inputs_to_GpuElemwise)
optdb.register('gpu_elemwise_fusion',
optdb.register('gpua_elemwise_fusion',
tensor.opt.FusionOptimizer(gpu_local_elemwise_fusion),
71.00, 'fast_run', 'fusion', 'local_elemwise_fusion', 'gpu')
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论